Update app.py

app.py

@@ -13,25 +13,57 @@ class DeepDreamInterpreter:
             
     def setup_client(self, api_key):
         """Set up the OpenAI client with the provided API key"""
-        self.api_key = api_key
-        self.client = OpenAI(
-            base_url="https://openrouter.ai/api/v1",
-            api_key=api_key,
-        )
-        return "API key configured successfully!"
+        if not api_key or not api_key.strip():
+            return "Please provide a valid API key."
+            
+        try:
+            self.api_key = api_key
+            self.client = OpenAI(
+                base_url="https://openrouter.ai/api/v1",
+                api_key=api_key,
+            )
+            # Test the API key with a simple request
+            test_response = self.client.chat.completions.create(
+                extra_headers={
+                    "HTTP-Referer": "https://deepdreaminterpreter.com",
+                    "X-Title": "Deep Dream Interpreter",
+                },
+                model="google/gemini-2.5-flash-preview:thinking",
+                messages=[
+                    {"role": "user", "content": "Hello, this is a test."}
+                ],
+                max_tokens=5  # Keep it minimal for the test
+            )
+            return "API key configured successfully!"
+        except Exception as e:
+            self.client = None
+            self.api_key = None
+            return f"API configuration failed: {str(e)}"
         
     def analyze_dream(self, dream_description, include_visualization=True, user_name="Anonymous"):
         """Analyze the provided dream description using Gemini 2.5 Flash"""
         if not self.client:
             return "Please configure your API key first."
             
-        if not dream_description.strip():
+        if not dream_description or not dream_description.strip():
             return "Please enter a dream description to analyze."
             
         try:
             # First pass: Dream analysis with thinking capabilities
             print("Starting dream analysis...")
-            analysis = self.client.chat.completions.create(
+            
+            # System prompt for dream analysis
+            system_prompt = """You are an expert dream analyst. Analyze the following dream description, identifying key symbols, emotions, themes, and potential psychological meanings. 
+            Structure your analysis in these sections: 
+            - Summary
+            - Key Symbols
+            - Emotional Landscape
+            - Themes
+            - Psychological Interpretation
+            
+            Be thoughtful, nuanced, and avoid overgeneralizations."""
+            
+            analysis_response = self.client.chat.completions.create(
                 extra_headers={
                     "HTTP-Referer": "https://deepdreaminterpreter.com",
                     "X-Title": "Deep Dream Interpreter",
@@ -40,7 +72,7 @@ class DeepDreamInterpreter:
                 messages=[
                     {
                         "role": "system",
-                        "content": "You are an expert dream analyst. Analyze the following dream description, identifying key symbols, emotions, themes, and potential psychological meanings. Structure your analysis in these sections: Summary, Key Symbols, Emotional Landscape, Themes, and Psychological Interpretation."
+                        "content": system_prompt
                     },
                     {
                         "role": "user",
@@ -49,7 +81,11 @@ class DeepDreamInterpreter:
                 ]
             )
             
-            dream_analysis = analysis.choices[0].message.content
+            # Check if we got a valid response
+            if not analysis_response or not hasattr(analysis_response, 'choices') or not analysis_response.choices:
+                return "Error: Received an invalid response from the API during analysis."
+                
+            dream_analysis = analysis_response.choices[0].message.content
             
             # If visualization is not needed, return just the analysis
             if not include_visualization:
@@ -57,7 +93,19 @@ class DeepDreamInterpreter:
                 
             # Second pass: Visual interpretation
             print("Generating visualization prompt...")
-            visualization_prompt = self.client.chat.completions.create(
+            
+            # System prompt for visualization
+            visualization_system_prompt = """Based on this dream analysis, create a detailed visual description that could be used as a prompt for an image generation model. 
+            Make it evocative and detailed, including:
+            - Colors and lighting
+            - Atmosphere and mood
+            - Composition and perspective
+            - Key elements and their arrangement
+            - Symbolic representations
+            
+            Create something that captures the essence and emotional quality of the dream."""
+            
+            visualization_response = self.client.chat.completions.create(
                 extra_headers={
                     "HTTP-Referer": "https://deepdreaminterpreter.com",
                     "X-Title": "Deep Dream Interpreter",
@@ -66,7 +114,7 @@ class DeepDreamInterpreter:
                 messages=[
                     {
                         "role": "system",
-                        "content": "Based on this dream analysis, create a detailed visual description that could be used as a prompt for an image generation model. Make it evocative and detailed, including colors, atmosphere, composition, and key elements."
+                        "content": visualization_system_prompt
                     },
                     {
                         "role": "user",
@@ -75,6 +123,18 @@ class DeepDreamInterpreter:
                 ]
             )
             
+            # Check if we got a valid response
+            if not visualization_response or not hasattr(visualization_response, 'choices') or not visualization_response.choices:
+                # If visualization fails, still return the analysis
+                return f"""## Dream Analysis
+
+{dream_analysis}
+
+## Visualization Prompt
+
+Error: Unable to generate visualization prompt. Please try again later.
+"""
+            
             # Combine the results
             full_response = f"""## Dream Analysis
 
@@ -82,7 +142,7 @@ class DeepDreamInterpreter:
 
 ## Visualization Prompt
 
-{visualization_prompt.choices[0].message.content}
+{visualization_response.choices[0].message.content}
 
 ---
 *Note: This visualization prompt can be used with image generation models like DALL-E, Midjourney, or Stable Diffusion to create a visual representation of your dream.*
@@ -90,34 +150,38 @@ class DeepDreamInterpreter:
             return full_response
             
         except Exception as e:
-            return f"Error: {str(e)}"
+            error_message = str(e)
+            print(f"Error during analysis: {error_message}")
+            return f"Error during analysis: {error_message}\n\nPlease check your API key and try again."
             
     def save_dream(self, user_name, dream_description, analysis):
         """Save the dream and its analysis to a JSON file"""
-        if not user_name.strip():
-            user_name = "Anonymous"
+        try:
+            if not user_name or not user_name.strip():
+                user_name = "Anonymous"
+                
+            # Create dreams directory if it doesn't exist
+            os.makedirs("dreams", exist_ok=True)
+                
+            # Create a filename based on user and timestamp
+            timestamp = time.strftime("%Y%m%d-%H%M%S")
+            filename = f"dreams/{user_name.replace(' ', '_')}_{timestamp}.json"
             
-        # Create dreams directory if it doesn't exist
-        if not os.path.exists("dreams"):
-            os.makedirs("dreams")
+            # Create the dream data
+            dream_data = {
+                "user": user_name,
+                "timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
+                "dream_description": dream_description,
+                "analysis": analysis
+            }
             
-        # Create a filename based on user and timestamp
-        timestamp = time.strftime("%Y%m%d-%H%M%S")
-        filename = f"dreams/{user_name.replace(' ', '_')}_{timestamp}.json"
-        
-        # Create the dream data
-        dream_data = {
-            "user": user_name,
-            "timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
-            "dream_description": dream_description,
-            "analysis": analysis
-        }
-        
-        # Save to file
-        with open(filename, "w") as f:
-            json.dump(dream_data, f, indent=4)
-            
-        return f"Dream saved to {filename}"
+            # Save to file
+            with open(filename, "w", encoding="utf-8") as f:
+                json.dump(dream_data, f, indent=4, ensure_ascii=False)
+                
+            return f"Dream saved to {filename}"
+        except Exception as e:
+            return f"Error saving dream: {str(e)}"
 
 
 # Set up the Gradio interface
@@ -127,17 +191,24 @@ def create_gradio_interface():
     
     # Define the process function for the API key
     def process_api_key(api_key):
-        if not api_key.strip():
+        if not api_key or not api_key.strip():
             return "Please enter a valid API key."
-        return interpreter.setup_client(api_key)
+        result = interpreter.setup_client(api_key)
+        return result
     
     # Define the process function for dream analysis
     def process_dream(api_key, dream_description, include_visualization, user_name):
-        if api_key.strip() != interpreter.api_key or not interpreter.client:
+        # If API key changed or not set up, configure it first
+        if not interpreter.client or api_key.strip() != interpreter.api_key:
             setup_message = interpreter.setup_client(api_key)
             if "successfully" not in setup_message:
                 return setup_message
         
+        # Validate dream description
+        if not dream_description or not dream_description.strip():
+            return "Please enter a dream description."
+        
+        # Perform analysis
         analysis = interpreter.analyze_dream(dream_description, include_visualization, user_name)
         return analysis
     
@@ -146,7 +217,7 @@ def create_gradio_interface():
         if not analysis_output or "Error" in analysis_output:
             return "Nothing to save or analysis contains errors."
         
-        if api_key.strip() != interpreter.api_key or not interpreter.client:
+        if not interpreter.client or api_key.strip() != interpreter.api_key:
             setup_message = interpreter.setup_client(api_key)
             if "successfully" not in setup_message:
                 return setup_message