yellowflash.ai

it is the main ai file

yellowflash.ai

@@ -0,0 +1,18 @@
+import gradio as gr, requests
+
+API_KEY="AIzaSyAPfDiu2V_aD6un00qHt5bkISm6C0Pkx7o"
+API_URL="https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent"
+
+def chat_fn(message, history):
+    try:
+        resp=requests.post(API_URL,headers={"Content-Type":"application/json",
+                                            "x-goog-api-key":API_KEY},
+                           json={"contents":[{"parts":[{"text":message}]}]},timeout=12)
+        ans=resp.json()["candidates"][0]["content"]["parts"][0]["text"]
+    except Exception as e:
+        ans=f"Error: {e}"
+    return ans
+
+demo = gr.ChatInterface(chat_fn, title="⚡ yellowflash.ai")
+demo.launch(share=True)
+