Update app.py

app.py

@@ -83,6 +83,13 @@ Settings.embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/al
 vector_store = ChromaVectorStore(chroma_collection=collection)
 index = VectorStoreIndex.from_vector_store(vector_store, embed_model=Settings.embed_model)
 query_engine = index.as_query_engine()
+
+def upload_file(filepath):
+    documents = SimpleDirectoryReader(filepath).load_data()
+    index = VectorStoreIndex.from_documents(documents)
+    query_engine = index.as_query_engine()
+    return filepath
+
 def rag(input_text, file):
     if (file):
 #        documents = []
@@ -107,30 +114,24 @@ 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):
-    documents = SimpleDirectoryReader(filepath).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("Pokreni")
-
-    u.upload(upload_file, u, file_out)
-    sub.click(rag, inp, out)
+#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("Pokreni")
+#
+#    u.upload(upload_file, u, file_out)
+#    sub.click(rag, inp, out)
     
 iface.launch()