lohitkolluri
diff --git a/‎.streamlit/config.toml
Lines changed: 3 additions & 3 deletions b/‎.streamlit/config.toml
Lines changed: 3 additions & 3 deletions
diff --git a/‎requirements.txt
Lines changed: 1 addition & 1 deletion b/‎requirements.txt
Lines changed: 1 addition & 1 deletion
diff --git a/‎src/api/LLM_Config.py
Lines changed: 1 addition & 19 deletions b/‎src/api/LLM_Config.py
Lines changed: 1 addition & 19 deletions
@@ -1,7 +1,7 @@
 [theme]
 base="dark"
 primaryColor="#00ADB5"
-backgroundColor="#222831"
-secondaryBackgroundColor="#393E46"
-textColor="#EEEEEE"
+backgroundColor="#1e1e1e"
+secondaryBackgroundColor="#333333"
+textColor="#64ffda"
 font="sans serif"
@@ -11,4 +11,4 @@ psycopg2-binary
 sqlalchemy
 # API and configuration
 python-dotenv
-openai
+google-generativeai
@@ -19,34 +19,16 @@ def get_completion_from_messages(
     temperature: float = 0.3,
 ) -> str:
     try:
-        # Combine system and user messages
         combined_message = f"{system_message}\n\nUser Query: {user_message}"
-        print("=== INPUT ===")
-        print(f"Combined Message:\n{combined_message}")
-        print(f"Temperature: {temperature}")
-        
+        # print(f"Combined Message:\n{combined_message}")
         model = genai.GenerativeModel(GEMINI_MODEL)
         response = model.generate_content(
             contents=combined_message,
             generation_config={"temperature": temperature}
         )
-
-        # Log raw response
-        print("\n=== RAW OUTPUT ===")
-        print(f"Response Object: {response}")
-
-        # Ensure text is a string and clean it
         text = response.text if isinstance(response.text, str) else str(response.text)
         clean_text = re.sub(r'```json\n|\n```', '', text)
-        
-        # Log cleaned text
-        print("\n=== CLEANED OUTPUT ===")
-        print(f"Cleaned Text:\n{clean_text}")
-        
         return clean_text
-
     except Exception as e:
         error_msg = f"Error generating response: {str(e)}"
-        print("\n=== ERROR ===")
-        print(error_msg)
         return error_msg