Update app.py

app.py

@@ -95,16 +95,17 @@ def rag(input_text, file):
         print("basename=", os.path.basename(file))
         print("filename=", file.name)        
         documents = SimpleDirectoryReader(file).load_data()
-        index = VectorStoreIndex.from_documents(documents)
+        index2 = VectorStoreIndex.from_documents(documents)
+        query_engine = index2.as_query_engine()
+        return query_engine.query(input_text)
 #        collection.add(
 #            documents=documents,
 #        ids=[f"id{last+i}" for i in range(len(documents))],
 #        metadatas=[{"state": "s0", "next": "s0", "used": False, "source": 'None', "page": -1, "lang": jezik } for i in range(len(documents)) ]
 #    )
-    query_engine = index.as_query_engine()
-    return query_engine.query(
-        input_text
-    )
+    else:
+        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)],