Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import gen 
+import psychohistory
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
@@ -42,22 +44,37 @@ def respond(
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
+
+
+with gr.Blocks(title="PSYCHOHISTORY") as app:
+
+
+    with gr.Tab("Search"):
+        with gr.Row():
+            txt_search = gr.Textbox(value="Iran and Israel war",label="Search Term",scale=5)
+            btn_search = gr.Button("Look",scale=1)
+        with gr.Row():
+            #search_results = gr.Dataframe(type="pandas")
+            mem_results = gr.JSON(label="Results")
+            btn_search.click(
+                gen.generate, 
+                inputs=[txt_search],
+                outputs=mem_results
+            )  
+            
+        #with gr.Row():
+        #    big_block = gr.HTML("""
+         #       <iframe style="scroll-padding-left: 50%; relative;background-color: #fff; height: 75vh; width: 100%; overflow-y: hidden; overflow-x: hidden;" src="https://holmesbot.com/api/shared?id=16657e456d9514"></iframe>
+         #   """)
+        
+    with gr.Tab("Graph"):
+        gr.load("models/stabilityai/stable-diffusion-xl-base-1.0")
+    with gr.Tab("Chat"):
+        
+        gr.ChatInterface(
+            respond,
+        )
 
 
 if __name__ == "__main__":
-    demo.launch()
+    app.launch()