Update app.py

app.py

@@ -1,3 +1,7 @@
+<<<<<<< HEAD
+
+=======
+>>>>>>> 2c4af7da413c1e35cee9d0757790fe2b61319a1e
 import gradio as gr
 import os
 from huggingface_hub import login
@@ -10,7 +14,7 @@ from chat.hermes_llm import ChatManager
 load_dotenv()
 
 # Login to Hugging Face
-login(token=os.getenv("HUGGINGFACEHUB_API_TOKEN"), add_to_git_credential=True)
+# login(token=os.getenv("HUGGINGFACEHUB_API_TOKEN"), add_to_git_credential=True)
 
 # Initialize components
 VECTOR_DB_PATH = os.getenv("VECTOR_DB_PATH")
@@ -25,10 +29,11 @@ def process_query(query, chat_history):
         if len(results)>0:
         # Get response from chat manager
             response = legal_chat.get_response(results[0]['content'], query=query)
+            response_final = f"""{response}\n\nsource:[case_id:{results[0]['metadata']['case_id']}]"""
         else :
-            response = "No Document match"
+            response_final = "No Document match"
         # Update chat history
-        chat_history.append((query, response))
+        chat_history.append((query, response_final))
         return "", chat_history
     except Exception as e:
         return "", chat_history + [(query, f"Error: {str(e)}")]