Update app.py

app.py

@@ -107,9 +107,31 @@ def rag(input_text, file):
         query_engine = index.as_query_engine()
         return query_engine.query(input_text)
 
-iface = gr.Interface(fn=rag, inputs=[gr.Textbox(label="Pitanje:", lines=6), gr.File()], 
-                     outputs=[gr.Textbox(label="Odgovor:", lines=6)], 
-                     title="Kako Vam mogu pomoći?",
-                     description= "UChat"
-                    )
+#iface = gr.Interface(fn=rag, inputs=[gr.Textbox(label="Pitanje:", lines=6), gr.File()], 
+#                     outputs=[gr.Textbox(label="Odgovor:", lines=6)], 
+#                     title="Kako Vam mogu pomoći?",
+#                     description= "UChat"
+#                    )
+
+def upload_file(filepath):
+    name = Path(filepath).name
+    documents = SimpleDirectoryReader(file).load_data()
+    index = VectorStoreIndex.from_documents(documents)
+    query_engine = index.as_query_engine()
+    return filepath
+
+with gr.Blocks() as iface:
+    gr.Markdown("Uchat")
+    file_out = gr.File()
+    with gr.Row():
+        with gr.Column(scale=1):
+            inp = gr.Textbox(label="Pitanje:", lines=6)
+            u = gr.UploadButton("Upload a file", file_count="single")
+        with gr.Column(scale=1):
+            out = gr.Textbox(label="Odgovor:", lines=6)
+            sub = gr.Button(label="Pokreni")
+
+    u.upload(upload_file, u, file_out)
+    sub.click(rag, inp, out)
+    
 iface.launch()