Update app.py

app.py

@@ -2,19 +2,47 @@ import gradio as gr
 import openai
 import os
 import base64
+from functools import lru_cache
+from PIL import Image
+import cv2
+import numpy as np
 
 # OpenAI API Configuration
 openai.api_key = os.getenv("OPENAI_API_KEY")
 ANALYSIS_MODEL = "gpt-4o"
 MAX_TOKENS = 4096
 
+# Persona Definitions
+PERSONAS = {
+    "Aggressive Trader": {
+        "description": "High-risk, short-term gains, leverages volatile market movements.",
+        "prompt": "Focus on high-risk strategies, short-term gains, and leverage opportunities. Suggest aggressive entry and exit points."
+    },
+    "Conservative Trader": {
+        "description": "Low-risk, long-term investments, prioritizes capital preservation.",
+        "prompt": "Focus on low-risk strategies, long-term investments, and capital preservation. Suggest safe entry points and strict stop-loss levels."
+    },
+    "Neutral Trader": {
+        "description": "Balanced approach, combines short and long-term strategies.",
+        "prompt": "Focus on balanced strategies, combining short and long-term opportunities. Suggest moderate risk levels and trend-following approaches."
+    },
+    "Reactive Trader": {
+        "description": "Quick decisions based on market news and social media trends.",
+        "prompt": "Focus on quick decision-making, momentum trading, and reacting to market news. Suggest strategies based on current trends and FOMO opportunities."
+    },
+    "Systematic Trader": {
+        "description": "Algorithm-based, rule-driven, and emotionless trading.",
+        "prompt": "Focus on algorithmic strategies, backtested rules, and quantitative analysis. Suggest data-driven entry and exit points."
+    }
+}
+
 # System Prompt Configuration
 SYSTEM_PROMPT = """Professional Crypto Technical Analyst:
 1. Identify all technical patterns in the chart
 2. Determine key support/resistance levels
 3. Analyze volume and momentum indicators
 4. Calculate risk/reward ratios
-5. Provide clear trading recommendations
+5. Provide clear trading recommendations based on the selected persona
 6. Include specific price targets
 7. Assess market sentiment
 8. Evaluate trend strength
@@ -22,6 +50,46 @@ SYSTEM_PROMPT = """Professional Crypto Technical Analyst:
 10. Provide time-based projections"""
 
 class ChartAnalyzer:
+    def __init__(self):
+        self.last_analysis = ""
+
+    def validate_image(self, image_path: str) -> bool:
+        """Validate if the uploaded file is a valid image and looks like a chart"""
+        try:
+            # Check if the file is an image
+            with Image.open(image_path) as img:
+                img.verify()  # Verify that it is a valid image
+
+            # Check if the image looks like a chart (basic check)
+            img = cv2.imread(image_path)
+            if img is None:
+                return False
+
+            # Simple check for chart-like features (e.g., axes, grid lines)
+            gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+            edges = cv2.Canny(gray, 50, 150)
+            if np.sum(edges) < 1000:  # Arbitrary threshold for edge detection
+                return False
+
+            return True
+        except Exception:
+            return False
+
+    def optimize_image(self, image_path: str) -> str:
+        """Optimize image size and quality"""
+        try:
+            img = Image.open(image_path)
+            # Resize if the image is too large
+            if img.size[0] > 1024 or img.size[1] > 1024:
+                img = img.resize((1024, 1024), Image.ANTIALIAS)
+            # Save optimized image
+            optimized_path = "optimized_chart.png"
+            img.save(optimized_path, "PNG", optimize=True, quality=85)
+            return optimized_path
+        except Exception as e:
+            print(f"Image optimization error: {str(e)}")
+            return image_path  # Fallback to original image
+
     def encode_image(self, image_path: str) -> str:
         """Encode image to base64 with validation"""
         if not os.path.exists(image_path):
@@ -34,17 +102,24 @@ class ChartAnalyzer:
         with open(image_path, "rb") as image_file:
             return base64.b64encode(image_file.read()).decode('utf-8')
 
-    def analyze_chart(self, image_path: str) -> str:
-        """Core analysis function"""
+    @lru_cache(maxsize=100)  # Cache up to 100 recent analyses
+    def analyze_chart(self, image_path: str, persona: str) -> str:
+        """Core analysis function with caching and persona-based recommendations"""
         try:
-            base64_image = self.encode_image(image_path)
+            # Optimize image before analysis
+            optimized_path = self.optimize_image(image_path)
+            base64_image = self.encode_image(optimized_path)
+            
+            # Add persona-specific instructions to the system prompt
+            persona_prompt = PERSONAS.get(persona, {}).get("prompt", "")
+            full_system_prompt = f"{SYSTEM_PROMPT}\n\n{persona_prompt}"
             
             response = openai.ChatCompletion.create(
                 model=ANALYSIS_MODEL,
                 messages=[
                     {
                         "role": "system",
-                        "content": SYSTEM_PROMPT
+                        "content": full_system_prompt
                     },
                     {
                         "role": "user",
@@ -196,12 +271,10 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
         gr.Markdown("""
         <div style="text-align: center; margin-bottom: 32px;">
             <h1 style="color: var(--primary-color); font-size: 2.5rem; margin-bottom: 16px;">
-                🧠 HundAI Q-Agent 
+                🚀 CryptoVision Pro
             </h1>
             <p style="color: var(--text-color); font-size: 1.1rem;">
-                Advanced Multifunctional HundAI-powered cryptocurrency technical analysis<br>
-                🔍 Deep Pattern Recognition<br>
-                🤖 Neural Market Forecasting
+                Advanced AI-powered cryptocurrency technical analysis
             </p>
         </div>
         """)
@@ -218,6 +291,11 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                         sources=["upload"],
                         height=300
                     )
+                    persona_dropdown = gr.Dropdown(
+                        choices=list(PERSONAS.keys()),
+                        label="Select Trading Persona",
+                        value="Neutral Trader"
+                    )
                     analyze_btn = gr.Button(
                         "Analyze Chart",
                         variant="primary",
@@ -236,12 +314,12 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
 
     # Event Handling
     analyze_btn.click(
-        fn=lambda: "Generating report...",  # Show loading text immediately
+        fn=lambda: loading_html,  # Show loading spinner immediately
         outputs=analysis_output,
         queue=False
     ).then(
         fn=analyzer.analyze_chart,
-        inputs=chart_input,
+        inputs=[chart_input, persona_dropdown],
         outputs=analysis_output,
         api_name="analyze"
     )