Update pages/10_Gemini-ChatOnly.py

pages/10_Gemini-ChatOnly.py

@@ -2,6 +2,7 @@ import streamlit as st
 import google.generativeai as genai
 
 # Set your API key
+
 api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"  # Replace with your actual API key
 
 genai.configure(api_key=api_key)
@@ -12,6 +13,7 @@ generation_config = genai.GenerationConfig(
     max_output_tokens=3000
 )
 
+
 # Initialize session state for chat history
 if 'chat_history' not in st.session_state:
     st.session_state['chat_history'] = []
@@ -28,7 +30,6 @@ def clear_conversation():
     st.session_state['chat_history'] = []
 
 # Send message function
-# Send message function with error handling
 def send_message():
     user_input = st.session_state.user_input
     if user_input:
@@ -38,22 +39,28 @@ def send_message():
 
         model = genai.GenerativeModel(
             model_name='gemini-pro',
-            generation_config=generation_config,
+            generation_config=generation_config
         )
 
         response = model.generate_content([{"role": "user", "parts": [{"text": chat_history_str}]}])
-
-        # Error handling for empty or blocked responses
-        if hasattr(response, 'prompt_feedback') and response.prompt_feedback:
-            st.error("The prompt was blocked or not processed correctly. Feedback: " + str(response.prompt_feedback))
-        elif not hasattr(response, 'parts') or not response.parts:
-            st.error("No response was generated. Please try again or modify your input.")
-        else:
-            # Assuming response.parts exists and has content
-            response_text = response.parts[0].text if hasattr(response.parts[0], "text") else "No response text found."
-            if response_text:
-                st.session_state['chat_history'].append({"role": "model", "parts":[{"text": response_text}]})
+        response_text = response.text if hasattr(response, "text") else "No response text found."
 
         st.session_state.user_input = ''
 
-    display_chat_history()
+    display_chat_history()
+
+# User input text area
+user_input = st.text_area(
+    "Enter your message here:",
+    value="",
+    key="user_input"
+)
+
+# Send message button
+send_button = st.button(
+    "Send",
+    on_click=send_message
+)
+
+# Clear conversation button
+clear_button = st.button("Clear Conversation", on_click=clear_conversation)