Update app.py

app.py

@@ -187,10 +187,20 @@ def log_conversation(timestamp, user_message, assistant_response, model_name, co
             api = HfApi(token=hf_token)
             filename = f"conversation_logs/daily_{current_date}.json"
             
+            # Check if the dataset repository exists, if not create it
+            try:
+                api.repo_info(repo_id="Mr-Geo/chroma_db", repo_type="dataset")
+            except Exception:
+                api.create_repo(
+                    repo_id="Mr-Geo/chroma_db",
+                    repo_type="dataset",
+                    private=True
+                )
+            
             try:
                 # Try to download existing file
                 existing_file = api.hf_hub_download(
-                    repo_id="Mr-Geo/bas_chat_logs",
+                    repo_id="Mr-Geo/chroma_db",
                     filename=filename,
                     repo_type="dataset",
                     token=hf_token
@@ -219,7 +229,7 @@ def log_conversation(timestamp, user_message, assistant_response, model_name, co
             ]
             
             api.create_commit(
-                repo_id="Mr-Geo/bas_chat_logs",
+                repo_id="Mr-Geo/chroma_db",
                 repo_type="dataset",
                 operations=operations,
                 commit_message=f"Update conversation logs for {current_date}"
@@ -258,6 +268,9 @@ def log_conversation(timestamp, user_message, assistant_response, model_name, co
 def chat_response(message, history, model_name):
     """Chat response function for Gradio interface"""
     try:
+        # Append 'at BAS' to the user's message
+        message += " at BAS"
+        
         # Get context and timestamp
         context = get_context(message)
         timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -294,7 +307,7 @@ Context: {context}"""
         # Add history and current message
         if history:
             for h in history:
-                messages.append({"role": "user", "content": f"{str(h[0])} at BAS"})
+                messages.append({"role": "user", "content": str(h[0])})
                 if h[1]:  # If there's a response
                     messages.append({"role": "assistant", "content": str(h[1])})
         
@@ -387,7 +400,7 @@ if __name__ == "__main__":
                 ],
                 value="llama-3.1-8b-instant",
                 label="Select AI Large Language Model 🤖",
-                info="Choose which AI model to use for responses (all models running on [GroqCloud](https://groq.com/groqrack/)"
+                info="Choose which AI model to use for responses (all models running on [GroqCloud](https://groq.com/groqrack/))"
             )
             
             chatbot = gr.Chatbot(height=600)