Update app.py

app.py

@@ -4,16 +4,16 @@ from getpass import getpass
 openai_api_key = os.getenv('OPENAI_API_KEY')
 openai_api_key = openai_api_key
 
+
 from llama_index.llms.openai import OpenAI
 from llama_index.embeddings.openai import OpenAIEmbedding
 from llama_index.core import Settings
 
-Settings.llm = OpenAI(model="gpt-3.5-turbo", temperature=0.4)
+Settings.llm = OpenAI(model="gpt-3.5-turbo",temperature=0.4)
 Settings.embed_model = OpenAIEmbedding(model="text-embedding-ada-002")
 
 from llama_index.core import SimpleDirectoryReader
 
-# Load initial documents
 documents = SimpleDirectoryReader("new_file").load_data()
 
 from llama_index.core import VectorStoreIndex, StorageContext
@@ -25,7 +25,7 @@ client = qdrant_client.QdrantClient(
 )
 
 vector_store = QdrantVectorStore(
-    collection_name="paper",
+    collection_name = "paper",
     client=client,
     enable_hybrid=True,
     batch_size=20,
@@ -51,40 +51,49 @@ chat_engine = index.as_chat_engine(
     memory=memory,
     system_prompt=(
         """You are an AI assistant who answers the user questions,
-           use the schema fields to generate appropriate and valid json queries"""
+           use the schema fields to generate appriopriate and valid json queries"""
     ),
 )
 
-import gradio as gr
+# def is_greeting(user_input):
 
+#     greetings = ["hello", "hi", "hey", "good morning", "good afternoon", "good evening", "greetings"]
+#     user_input_lower = user_input.lower().strip()
+#     return any(greet in user_input_lower for greet in greetings)
+# def is_bye(user_input):
+
+#     greetings = ["thanks", "thanks you", "thanks a lot", "good answer", "good bye", "bye bye"]
+#     user_input_lower = user_input.lower().strip()
+#     return any(greet in user_input_lower for greet in greetings)
+import gradio as gr
 def chat_with_ai(user_input, chat_history):
+    # if is_greeting(user_input):
+    #     response = 'hi, how can i help you?'
+    #     chat_history.append((user_input, response))
+    #     return chat_history, ""
+    # elif is_bye(user_input):
+    #     response = "you're wlocome"
+    #     chat_history.append((user_input, response))
+    #     return chat_history, ""
     response = chat_engine.chat(user_input)
     references = response.source_nodes
-    ref, pages = [], []
+    ref,pages = [],[]
     for i in range(len(references)):
-        if references[i].metadata['file_name'] not in ref:
-            ref.append(references[i].metadata['file_name'])
-    complete_response = str(response) + "\n\n"
-    if ref != [] or pages != []:
-        chat_history.append((user_input, complete_response))
-        ref = []
-    elif ref == [] or pages == []:
-        chat_history.append((user_input, str(response)))
+      if references[i].metadata['file_name'] not in ref:
+        ref.append(references[i].metadata['file_name'])
+      # pages.append(references[i].metadata['page_label'])
+    complete_response = str(response) + "\n\n" 
+    if ref !=[] or pages!=[]:
+      chat_history.append((user_input, complete_response))
+      ref = []
+    elif ref==[] or pages==[]:
+      chat_history.append((user_input,str(response)))
         
     return chat_history, ""
 
 def clear_history():
     return [], ""
 
-def upload_file(file):
-    # Save the uploaded file to the "new_file" directory
-    if not os.path.exists("new_file"):
-        os.makedirs("new_file")
-    file_path = os.path.join("new_file", file.name)
-    with open(file_path, "wb") as f:
-        f.write(file.read())
-    return f"File {file.name} uploaded successfully!"
-
 def gradio_chatbot():
     with gr.Blocks() as demo:
         gr.Markdown("# Chat Interface for LlamaIndex")
@@ -95,24 +104,15 @@ def gradio_chatbot():
         )
 
         submit_button = gr.Button("Send")
-        btn_clear = gr.Button("Delete Context")
-
-        # Add a file upload component
-        file_upload = gr.File(label="Upload a file")
+        btn_clear = gr.Button("Delete Context") 
 
-        # Add a button to handle file upload
-        upload_button = gr.Button("Upload File")
 
         chat_history = gr.State([])
 
-        # Define the file upload action
-        upload_button.click(upload_file, inputs=file_upload, outputs=user_input)
-
-        # Define the chat interaction
         submit_button.click(chat_with_ai, inputs=[user_input, chat_history], outputs=[chatbot, user_input])
 
         user_input.submit(chat_with_ai, inputs=[user_input, chat_history], outputs=[chatbot, user_input])
-        btn_clear.click(fn=clear_history, outputs=[chatbot, user_input])
+        btn_clear.click(fn=clear_history, outputs=[chatbot, user_input])  
 
     return demo