Update ap.py

ap.py

@@ -2,118 +2,150 @@ import gradio as gr
 import os
 from openai import OpenAI
 
-client = OpenAI(
-    api_key=os.environ.get("ANTHROPIC_API_KEY", ""),
-    base_url="https://api.avalai.ir/v1"
-)
+API_KEY = os.environ.get("AVALAI_API_KEY", "")
+BASE_URL = "https://api.avalai.ir/v1"
+MODEL = "claude-sonnet-4-6"
 
-API_KNOWLEDGE_BASE = """
-You are an expert API architect and indie hacker advisor. You know every major API deeply:
+SYSTEM_PROMPT = """You are an expert API architect and indie hacker advisor.
+
+You know every major API deeply:
 
 PAYMENT: Stripe, Paddle, LemonSqueezy, PayPal, Razorpay
-AI/ML: OpenAI, Anthropic Claude, Replicate, HuggingFace Inference, AssemblyAI, ElevenLabs, Stability AI
-COMMUNICATION: Twilio (SMS/Voice), SendGrid (email), Resend, Mailgun, Vonage, WhatsApp Business
-DATA/ENRICHMENT: Clearbit, Apollo.io, Hunter.io, Proxycurl, PeopleDataLabs
-MAPS/LOCATION: Google Maps, Mapbox, HERE, OpenStreetMap/Nominatim
+AI/ML: OpenAI, Anthropic Claude, Replicate, HuggingFace Inference, AssemblyAI, ElevenLabs
+COMMUNICATION: Twilio (SMS/Voice), SendGrid (email), Resend, Mailgun, WhatsApp Business
+DATA/ENRICHMENT: Clearbit, Apollo.io, Hunter.io, Proxycurl
+MAPS/LOCATION: Google Maps, Mapbox, HERE, OpenStreetMap
 MEDIA: Cloudinary, Mux (video), Imgix, Transloadit
-FINANCE/MARKET DATA: Alpha Vantage, Polygon.io, CoinGecko, Plaid, Finnhub
-SEARCH: Algolia, Typesense, Elasticsearch, Meilisearch
+FINANCE: Alpha Vantage, Polygon.io, CoinGecko, Plaid, Finnhub
+SEARCH: Algolia, Typesense, Meilisearch
 AUTH: Auth0, Clerk, Supabase Auth, Firebase Auth
-DATABASE/BACKEND: Supabase, Firebase, PlanetScale, Neon, Upstash
-SCRAPING/CRAWLING: Apify, ScraperAPI, Browserless, Firecrawl
-SOCIAL: Twitter/X API, Reddit API, LinkedIn API, Instagram Graph API
-PRODUCTIVITY: Notion API, Airtable, Google Workspace, Microsoft Graph
-E-COMMERCE: Shopify, WooCommerce, Printful (print-on-demand)
+DATABASE: Supabase, Firebase, PlanetScale, Neon, Upstash
+SCRAPING: Apify, ScraperAPI, Browserless, Firecrawl
 ANALYTICS: Mixpanel, PostHog, Amplitude, Plausible
-"""
 
-SYSTEM_PROMPT = API_KNOWLEDGE_BASE + """
-
-When a user describes their product idea, you will recommend the perfect API stack.
+When a user describes their product idea, recommend the perfect API stack.
 
 Your response MUST follow this exact format:
 
 ## 🎯 Your Idea in One Line
-(restate the idea clearly and concisely)
+(restate the idea clearly)
 
 ## 🔧 Recommended API Stack
 
-For each API (recommend 3-5 total), use this format:
+For each API (recommend 3-5 total):
 
 ### [API Name] — [Role in the product]
 - **What it does for you:** (one sentence)
-- **Pricing:** (free tier + paid tier, be specific)
+- **Pricing:** (free tier + paid, be specific with numbers)
 - **Docs:** (exact URL)
 - **Why not alternatives:** (one sentence)
 - **Integration difficulty:** Easy / Medium / Hard
 
-## 💰 Total Monthly API Cost Estimate
-(breakdown for 100 users / 1000 users / 10,000 users)
+## 💰 Monthly Cost Estimate
+| Users | Est. Cost |
+|-------|-----------|
+| 100   | $X/month  |
+| 1,000 | $X/month  |
+| 10,000| $X/month  |
 
 ## ⚡ Build Order
-(which API to integrate first, second, third — and why)
+1. First: [API] — because...
+2. Second: [API] — because...
+3. Third: [API] — because...
 
-## 🚨 One Thing to Watch Out For
+## 🚨 Top Mistake to Avoid
 (the most common mistake with this stack)
 
-Be specific, honest about pricing, and always mention free tiers.
-"""
+Be specific, honest about pricing, always mention free tiers."""
 
-def get_stack_recommendation(idea, budget, technical_level, target_market):
-    if not idea.strip():
-        return "Please describe your product idea first."
-
-    if not client.api_key:
-        return "⚠️ ANTHROPIC_API_KEY is not set. Add it in Space Settings → Secrets."
 
-    user_message = f"""
-My product idea: {idea}
+def get_client():
+    return OpenAI(
+        api_key=API_KEY,
+        base_url=BASE_URL
+    )
 
-My monthly API budget: {budget}
-My technical level: {technical_level}
-Target market: {target_market}
 
-Please recommend the best API stack for this.
-"""
+def get_stack(idea, budget, level, market):
+    if not idea.strip():
+        return "⚠️ Please describe your product idea."
+
+    if not API_KEY:
+        return (
+            "⚠️ **AVALAI_API_KEY is not set.**\n\n"
+            "Go to: **Space Settings → Secrets → New secret**\n"
+            "- Name: `AVALAI_API_KEY`\n"
+            "- Value: your key from [avalai.ir](https://avalai.ir)"
+        )
 
     try:
+        client = get_client()
+        prompt = f"""Product idea: {idea}
+
+Monthly API budget: {budget}
+Technical level: {level}
+Target market: {market}
+
+Recommend the best API stack for this product."""
+
         response = client.chat.completions.create(
-            model="claude-sonnet-4-6",
-            max_tokens=1500,
+            model=MODEL,
             messages=[
                 {"role": "system", "content": SYSTEM_PROMPT},
-                {"role": "user", "content": user_message}
-            ]
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=1500,
+            temperature=0.7
         )
         return response.choices[0].message.content
+
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        err = str(e)
+        if "401" in err or "auth" in err.lower():
+            return "❌ Invalid API key. Check your AVALAI_API_KEY in Space Settings."
+        if "429" in err:
+            return "❌ Rate limit hit. Please wait a moment and try again."
+        if "model" in err.lower():
+            return f"❌ Model error — check if `claude-sonnet-4-6` is available on your AvалAI plan.\n\nDetails: {err}"
+        return f"❌ Error: {err}"
 
-def ask_followup(question, previous_recommendation):
+
+def ask_followup(question, prev):
     if not question.strip():
         return ""
-    if not previous_recommendation or previous_recommendation.startswith("Please") or previous_recommendation.startswith("⚠️"):
-        return "Please generate a stack recommendation first."
+    if not prev or prev.startswith("⚠️") or prev.startswith("❌"):
+        return "⚠️ Please generate a stack recommendation first."
 
     try:
+        client = get_client()
         response = client.chat.completions.create(
-            model="claude-sonnet-4-6",
-            max_tokens=800,
+            model=MODEL,
             messages=[
                 {"role": "system", "content": SYSTEM_PROMPT},
-                {"role": "user", "content": f"I got this API stack recommendation:\n\n{previous_recommendation}\n\nMy follow-up question: {question}"}
-            ]
+                {
+                    "role": "user",
+                    "content": (
+                        f"Previous recommendation:\n{prev}\n\n"
+                        f"Follow-up question: {question}\n\n"
+                        f"Answer concisely and helpfully."
+                    )
+                }
+            ],
+            max_tokens=800,
+            temperature=0.7
         )
         return response.choices[0].message.content
+
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
+
 EXAMPLES = [
-    ["A Notion-like note taking app with AI summarization", "$50/month", "Intermediate", "Students and researchers"],
-    ["An SMS marketing platform for small restaurants", "$30/month", "Beginner", "Local restaurant owners"],
-    ["A job board that auto-matches candidates with AI", "$100/month", "Advanced", "Tech recruiters"],
-    ["A podcast transcription and highlight tool", "$20/month", "Beginner", "Podcasters"],
-    ["A crypto portfolio tracker with price alerts", "$0 (free only)", "Intermediate", "Retail crypto investors"],
+    ["A Notion-like notes app with AI summarization", "$50/month", "Intermediate", "Students"],
+    ["SMS marketing platform for small restaurants", "$30/month", "Beginner", "Restaurant owners"],
+    ["Job board that auto-matches candidates with AI", "$100/month", "Advanced", "Tech recruiters"],
+    ["Podcast transcription and highlight tool", "$20/month", "Beginner", "Podcasters"],
+    ["Crypto portfolio tracker with price alerts", "$0 (free only)", "Intermediate", "Retail investors"],
 ]
 
 css = """
@@ -124,56 +156,61 @@ footer { display: none !important; }
 with gr.Blocks(css=css, title="API Stack Finder") as demo:
     gr.Markdown("""
 # 🔧 API Stack Finder
-**Describe your product idea → Get the perfect API stack, pricing breakdown, and build order**
+**Describe your product idea → Get the perfect API stack, pricing, and build order**
 
-Powered by Claude (Anthropic) — No more guessing which APIs to use.
+Powered by **Claude Sonnet 4.6** via [AvалAI](https://avalai.ir)
 """)
 
     with gr.Row():
         with gr.Column(scale=3):
             idea_input = gr.Textbox(
                 label="Describe your product idea",
-                placeholder="e.g. A platform where freelancers can sell their services and get paid instantly via Stripe...",
+                placeholder="e.g. A platform where freelancers can sell services and get paid instantly...",
                 lines=4
             )
         with gr.Column(scale=1):
             budget_input = gr.Dropdown(
-                choices=["$0 (free only)", "$10/month", "$30/month", "$50/month", "$100/month", "$500/month", "Unlimited"],
+                choices=[
+                    "$0 (free only)",
+                    "$10/month",
+                    "$30/month",
+                    "$50/month",
+                    "$100/month",
+                    "$500/month",
+                    "Unlimited"
+                ],
                 value="$30/month",
                 label="Monthly API budget"
             )
             level_input = gr.Dropdown(
                 choices=["Beginner", "Intermediate", "Advanced"],
                 value="Intermediate",
-                label="Your technical level"
+                label="Technical level"
             )
             market_input = gr.Textbox(
                 label="Target market",
-                placeholder="e.g. Small business owners in Europe",
+                placeholder="e.g. Small business owners",
                 value="General consumers"
             )
 
     gr.Markdown("**Try an example:**")
     with gr.Row():
         for ex in EXAMPLES:
-            gr.Button(ex[0][:35] + "...", size="sm").click(
+            gr.Button(ex[0][:32] + "...", size="sm").click(
                 lambda e=ex: (e[0], e[1], e[2], e[3]),
                 outputs=[idea_input, budget_input, level_input, market_input]
             )
 
     submit_btn = gr.Button("🔍 Find My API Stack", variant="primary", size="lg")
-
-    recommendation_output = gr.Markdown(
-        value="*Your API stack recommendation will appear here...*",
-        label="Recommended Stack"
-    )
+    output = gr.Markdown("*Your API stack will appear here...*")
 
     gr.Markdown("---")
-    gr.Markdown("### 💬 Ask a follow-up question")
+    gr.Markdown("### 💬 Ask a follow-up")
+
     with gr.Row():
         followup_input = gr.Textbox(
-            label="Follow-up",
-            placeholder="e.g. Is there a cheaper alternative to Twilio? / How do I handle auth without Auth0?",
+            label="Follow-up question",
+            placeholder="e.g. Is there a free alternative to Stripe? / How do I add auth cheaply?",
             scale=4
         )
         followup_btn = gr.Button("Ask", scale=1, variant="secondary")
@@ -181,21 +218,19 @@ Powered by Claude (Anthropic) — No more guessing which APIs to use.
     followup_output = gr.Markdown()
 
     submit_btn.click(
-        get_stack_recommendation,
+        get_stack,
         inputs=[idea_input, budget_input, level_input, market_input],
-        outputs=recommendation_output
+        outputs=output
     )
-
     followup_btn.click(
         ask_followup,
-        inputs=[followup_input, recommendation_output],
+        inputs=[followup_input, output],
         outputs=followup_output
     )
 
     gr.Markdown("""
 ---
-Built with [Gradio](https://gradio.app) + [Claude](https://anthropic.com) · 
-[View on HuggingFace](https://huggingface.co/spaces/Alireza1913/Monetizable_API_Finder)
+🤖 Model: `claude-sonnet-4-6` · 🔑 API: [avalai.ir](https://avalai.ir)
 """)
 
 if __name__ == "__main__":