Update app.py

app.py

@@ -2,87 +2,79 @@ import os
 import gradio as gr
 import requests
 
-# -----------------------------
-# GPT5 Model Class
-# -----------------------------
 class GPT5Model:
-    def __init__(self, api_key):
+    """
+    GPT5Model handles interactions with the GPT-5 API.
+    It includes system prompts, request construction, timeout handling,
+    and graceful error responses if the API can't be reached.
+    """
+    def __init__(self, api_key: str):
         self.api_key = api_key
         self.system_prompt = (
             "You are GPT-5, the most advanced AI model available. "
             "Answer accurately, intelligently, and helpfully."
         )
 
-    def generate_response(self, prompt):
+    def generate_response(self, prompt: str) -> str:
+        """
+        Sends a prompt to the GPT-5 API and returns the response text.
+        If there's a connection error, timeout, or invalid API key, returns a friendly error message.
+        """
+        url = "https://api.pplx.ai/v1/generate"
         headers = {
             "Authorization": f"Bearer {self.api_key}",
             "Content-Type": "application/json"
         }
         full_prompt = f"{self.system_prompt}\nUser: {prompt}\nGPT-5:"
-        data = {
-            "prompt": full_prompt,
-            "max_tokens": 500
-        }
+        payload = {"prompt": full_prompt, "max_tokens": 500}
 
         try:
-            response = requests.post("https://api.pplx.ai/v1/generate", json=data, headers=headers, timeout=15)
+            response = requests.post(url, json=payload, headers=headers, timeout=15)
+        except requests.exceptions.Timeout:
+            return "Error: Request timed out. Please check your network or try again later."
+        except requests.exceptions.ConnectionError:
+            return "Error: Could not reach API. Please check network settings."
         except requests.exceptions.RequestException as e:
-            return f"Error: Could not reach API. {e}"
+            return f"Error: Unexpected error occurred: {e}"
 
+        if response.status_code == 401:
+            return "Error: API key invalid or expired."
         if response.status_code != 200:
-            return f"Error: API key exhausted or invalid. ({response.status_code})"
+            return f"Error: API returned status {response.status_code}."
 
         try:
-            result = response.json()
-            choices = result.get("choices", [])
-            if not choices or "text" not in choices[0] or not choices[0]["text"].strip():
-                return "Error: API key exhausted or invalid."
-            return choices[0]["text"].strip()
-        except Exception:
-            return "Error: API key exhausted or invalid."
+            data = response.json()
+            text = data.get("choices", [{}])[0].get("text", "").strip()
+            if not text:
+                return "Error: No response content. API key may be exhausted."
+            return text
+        except ValueError:
+            return "Error: Could not parse API response."
 
-# -----------------------------
-# Load API key
-# -----------------------------
+# === Load API Key ===
 api_key = os.getenv("PPLX_API_KEY")
 if not api_key:
-    raise ValueError("API key not found. Please set PPLX_API_KEY environment variable.")
+    raise EnvironmentError("API key not found. Please set PPLX_API_KEY environment variable.")
 
 model = GPT5Model(api_key)
 
-# -----------------------------
-# UI Function
-# -----------------------------
-def chat_with_gpt5(user_input):
-    return model.generate_response(user_input)
+def respond(message, chat_history):
+    reply = model.generate_response(message)
+    chat_history.append((message, reply))
+    return "", chat_history
 
-# -----------------------------
-# Build Better UI
-# -----------------------------
+# === Gradio UI ===
 with gr.Blocks(css="""
-    #title {text-align: center; font-size: 28px; font-weight: bold;}
-    #footer {text-align: center; font-size: 14px; color: gray;}
+     #title {text-align: center; font-size: 28px; font-weight: bold;}
+     #footer {text-align: center; font-size: 14px; color: gray;}
 """) as demo:
     gr.Markdown("<div id='title'>🚀 GPT-5 Model Interface</div>")
+    chatbot = gr.Chatbot(label="Conversation with GPT-5", height=400)
     with gr.Row():
-        with gr.Column(scale=1):
-            chatbot = gr.Chatbot(height=400, label="GPT-5 Conversation")
-            with gr.Row():
-                txt = gr.Textbox(
-                    show_label=False,
-                    placeholder="Type your message here...",
-                    container=False
-                )
-                send_btn = gr.Button("Send", variant="primary")
-
+        txt = gr.Textbox(show_label=False, placeholder="Type your message here…", container=False)
+        send_btn = gr.Button("Send", variant="primary")
     gr.Markdown("<div id='footer'>Powered by GPT-5 API Simulation | © 2025</div>")
 
-    # Handle chat logic
-    def respond(message, chat_history):
-        bot_reply = chat_with_gpt5(message)
-        chat_history.append(("You: " + message, "GPT-5: " + bot_reply))
-        return "", chat_history
-
     send_btn.click(respond, [txt, chatbot], [txt, chatbot])
     txt.submit(respond, [txt, chatbot], [txt, chatbot])