Update app.py

app.py

@@ -62,10 +62,10 @@ def chat_query(question, history):
 
     
     query = f"""As an experienced Electrical Engineer, please provide an elaborate, precise, and answer politely pointwise to the question: {question}.
-              Also, Please consider the provided chat history: {chat_history}
-              Ensure that your response is detailed, accurate, and addresses each aspect of the question thoroughly.
-              If the context of the question doesn't align with the chat history, please provide your response in a fresh manner.
-              If don't get the answer, feel free to reply from Internet."""
+              Also, Please consider the provided chat history: {history}.
+              Ensure that your current response is detailed, accurate, and addresses each aspect of the question thoroughly.
+              If the context of the question doesn't align with your last reply, please provide your response in a fresh manner.
+              If don't get the answer, feel free to reply from your own knowledge."""         
 
     #llm = ChatOpenAI(model=llm_name, temperature=0.1, api_key = OPENAI_API_KEY)
     llm = GoogleGenerativeAI(model="gemini-1.0-pro", google_api_key=GEMINI_API_KEY)
@@ -78,17 +78,20 @@ def chat_query(question, history):
     # Replace input() with question variable for Gradio
     result = qa({"question": query, "chat_history" : chat_history})
 
-    
     source_docs = result["source_documents"]
-    #file_name = get_file(source_docs)[0]
-    file_name = os.path.basename(source_docs[0].metadata['source'])
+    file_names = get_file(source_docs)
+    #file_name = os.path.basename(source_docs[0].metadata['source'])
+    file_name = ', '.join([f"{x}" for x in file_names[:3]])
+
+    # print("History : ", history)
+    # print("\n Chat_his : ", chat_history)
 
-    return result["answer"] + "\n\nSource : " + file_name
+    return result["answer"] + "\n\nSources : " + file_name
 
 
 iface = gr.ChatInterface(
     fn=chat_query,
-    #title=" -----:  Conversational BOT for Model-TS  :----- ",
+    #title=" --:  Conversational BOT for Model-TS  :----- ",
     description="""-- Welcome to the Language Model trained on `Model Technical Specifications` of SS-Engg. Dept.\n\n
                    -- The Model tries to answer the Query based on TS, GTR & Specific Requirements. \n\n
                    -- For precise reply, please input `Specific Keywords` in your Query. \n\n """,
@@ -104,15 +107,17 @@ iface = gr.ChatInterface(
 
 )
 
-Title= "# -----:  Conversational BOT for Model-TS  :----- "
+Title= "# --:  Conversational BOT for Model-TS  :-- "
 
 
 with gr.Blocks(css="style.css", fill_height=True) as demo:
     with gr.Row():
       with gr.Column(scale=1):
-        gr.Image("PG Logo.png", width=50, show_download_button = False, show_label = False, show_share_button = False )
+        gr.Image("Chatbot.png", width=50, show_download_button = False, show_label = False, show_share_button = False )
       with gr.Column(scale=10):
         gr.Markdown(Title)
+      with gr.Column(scale=1):
+        gr.Image("PG Logo.png", width=50, show_download_button = False, show_label = False, show_share_button = False )  
     
     iface.render()