Better UI

app.py

@@ -126,25 +126,41 @@ def print_array(arr):
 def html_link_generator(book, page):
     bookurl = book_url(book)
     url = f"https://diagrams1.s3.ap-south-1.amazonaws.com/anatomybooks/{bookurl}#page={page}"
-    html = f'<iframe src="{url}" width="800" height="600"></iframe>'
+    # html = f'<iframe src="{url}" width="800" height="600"></iframe>'
     # print(url)
-    return html
+    
+    return url
 
 def getanswer(question):
+    if question=="" :
+        return "Please ask a question" , []
     llm_response = qa_chain(question)
     response = process_llm_response(llm_response)
-    sources= print_array(response["sources"])
     html= html_link_generator(response["sources"][0]["book"][22:], response["sources"][0]["page"])
     # html = """<iframe src="https://diagrams1.s3.ap-south-1.amazonaws.com/anatomybooks/BD+Chaurasia's+Handbook+of+General+Anatomy%2C+4th+Edition.pdf#page=40" width="800" height="600"></iframe>"""
-    return response["result"], sources, html
+    return response["result"], response['sources']
 
+def makevisible(source1,source2,source3):
+    return{
+        source1: gr.update(visible=True),
+        source2: gr.update(visible=True),
+        source3: gr.update(visible=True)
+    }
 
-demo = gr.Interface(
-    fn=getanswer,
-    inputs=["text"],
-    outputs=["text", "text", "html"],
-    api="question",
-    title="Medihelp Chatbot",
 
-)
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column(scale=1, min_width=600):
+            question = gr.Textbox(label="Question")
+            submitbtn = gr.Button("Submit").style(full_width=True)
+            answer = gr.Textbox(label="Answer", interactive=False)
+            sources = gr.Json(label="Sources", interactive=False)
+            source1 = gr.Button(label="Source 1", visible=False)
+            source2 = gr.Button(label="Source 2", visible=False)
+            source3 = gr.Button(label="Source 3", visible=False)
+            
+            submitbtn.click(fn=getanswer, inputs=[question], outputs=[answer, sources], api_name="question")
+            # source1.click(fn=None, _js=f"""window.open('"""+"""', target="_blank");""")            
+            # sources.change(make_source_buttons, [sources, source1, source2, source3], [source1,source2,source3])
+            
 demo.launch()