Update app.py

app.py

@@ -5,6 +5,7 @@ from dotenv import load_dotenv
 import json
 import textwrap
 import time  # For handling potential rate limits
+from typing import Any, Optional
 
 # Load environment variables
 load_dotenv()
@@ -39,7 +40,7 @@ with st.sidebar:
         -   **Clear Instructions**:  Provide explicit and unambiguous directions.
         -   **Delimiters**:   Use special characters to separate input parts.
         -   **Structured Output**: Request output in a specific format (JSON).
-        -   **Assumption Checking**: Verify conditions in the input.
+        -   **Assumption Checking**:  Verify conditions in the input.
         -   **Few-Shot Prompting**: Provide input-output examples.
         -   **Temperature Control**: Adjust output randomness.
         -   **Chain of Thought (CoT)**: Elicit step-by-step reasoning.
@@ -72,7 +73,7 @@ def code_block(text: str, language: str = "text") -> None:
     st.markdown(f"```{language}\n{text}\n```", unsafe_allow_html=True)
 
 
-def display_response(response: genai.GenerateContentResponse) -> None:
+def display_response(response: Any) -> None: # Removed the type hint GenerateContentResponse
     """Displays the model's response, handling text, and error cases."""
     if response.text:
         st.subheader("Generated Response:")
@@ -84,7 +85,7 @@ def display_response(response: genai.GenerateContentResponse) -> None:
         st.error(f"Full response object: {response}")  # Print the full response for debugging
 
 
-def generate_with_retry(prompt: str, generation_config: genai.types.GenerationConfig, max_retries: int = 3, delay: int = 5) -> genai.GenerateContentResponse:
+def generate_with_retry(prompt: str, generation_config: genai.types.GenerationConfig, max_retries: int = 3, delay: int = 5) -> Any: # Removed the type hint
     """
     Generates content with retry logic to handle potential API errors (e.g., rate limits).
 
@@ -254,4 +255,4 @@ if st.button("Generate Response"):
                     display_response(response)
 
             except Exception as e:
-                st.error(f"An error occurred: {e}")
+                st.error(f"An error occurred: {e}")