Create app.py

app.py

@@ -0,0 +1,193 @@
+import gradio as gr
+import requests
+import base64
+import os
+
+# --- UI Constants ---
+TITLE = "🤖 Chatbot based on Responses API by Pejman Ebrahimi"
+FOOTER = "@2025 Pejman Ebrahimi - All rights reserved."
+LINKS = [
+    ("[![LinkedIn](https://img.shields.io/badge/LinkedIn-0077B5?style=for-the-badge&logo=linkedin&logoColor=white)](https://www.linkedin.com/in/pejman-ebrahimi-4a60151a7/)", "https://www.linkedin.com/in/pejman-ebrahimi-4a60151a7/"),
+    ("[![HuggingFace](https://img.shields.io/badge/🤗_Hugging_Face-FFD21E?style=for-the-badge)](https://huggingface.co/arad1367)", "https://huggingface.co/arad1367"),
+    ("[![Website](https://img.shields.io/badge/Website-008080?style=for-the-badge&logo=About.me&logoColor=white)](https://arad1367.github.io/pejman-ebrahimi/)", "https://arad1367.github.io/pejman-ebrahimi/"),
+    ("[![University](https://img.shields.io/badge/University-00205B?style=for-the-badge&logo=academia&logoColor=white)](https://www.uni.li/pejman.ebrahimi?set_language=en)", "https://www.uni.li/pejman.ebrahimi?set_language=en"),
+]
+
+# --- Helper Functions ---
+
+def check_api_key(api_key):
+    """Check if the provided OpenAI API key is valid by making a minimal request."""
+    try:
+        url = "https://api.openai.com/v1/models"
+        headers = {"Authorization": f"Bearer {api_key}"}
+        r = requests.get(url, headers=headers, timeout=10)
+        return r.status_code == 200
+    except Exception:
+        return False
+
+def openai_responses_api(api_key, messages, previous_response_id=None):
+    """Call OpenAI Responses API for chat or image analysis."""
+    url = "https://api.openai.com/v1/responses"
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "model": "gpt-4o-mini",
+        "input": messages
+    }
+    if previous_response_id:
+        payload["previous_response_id"] = previous_response_id
+    r = requests.post(url, headers=headers, json=payload, timeout=60)
+    if r.status_code == 200:
+        return r.json()
+    else:
+        raise Exception(r.json().get("error", {}).get("message", "Unknown error"))
+
+def encode_image_to_base64(image):
+    """Encode uploaded image to base64 string."""
+    if image is None:
+        return None
+    with open(image, "rb") as f:
+        return base64.b64encode(f.read()).decode("utf-8")
+
+def get_iframe_code(space_url):
+    """Generate iframe code for embedding."""
+    return f'<iframe src="{space_url}" width="100%" height="600" frameborder="0"></iframe>'
+
+# --- Gradio App Logic ---
+
+def set_api_key(api_key, state):
+    """Set and validate API key in session state."""
+    if not api_key or not api_key.startswith("sk-"):
+        return gr.update(visible=True), "Please enter a valid OpenAI API key.", state
+    if check_api_key(api_key):
+        state["api_key"] = api_key
+        return gr.update(visible=False), "", state
+    else:
+        return gr.update(visible=True), "API key is invalid. Please try again.", state
+
+def chat_fn(user_message, chat_history, image, state):
+    """Main chat function: handles text and optional image."""
+    api_key = state.get("api_key")
+    if not api_key:
+        return chat_history, state, gr.update(visible=True), "Please enter your OpenAI API key."
+    # Prepare messages for API
+    messages = []
+    for turn in chat_history:
+        messages.append({"role": "user", "content": turn[0]})
+        messages.append({"role": "assistant", "content": turn[1]})
+    # Add current user message
+    if image is not None:
+        # Image attached: send as multimodal input
+        base64_img = encode_image_to_base64(image)
+        content = [
+            {"type": "input_text", "text": user_message or "Please describe this image."},
+            {"type": "input_image", "image_url": f"data:image/jpeg;base64,{base64_img}"}
+        ]
+        messages.append({"role": "user", "content": content})
+    else:
+        messages.append({"role": "user", "content": user_message})
+    # Call OpenAI Responses API
+    try:
+        response = openai_responses_api(api_key, messages)
+        output_text = response.get("output_text", "")
+        chat_history.append((user_message, output_text))
+        state["previous_response_id"] = response.get("id")
+        return chat_history, state, gr.update(visible=False), ""
+    except Exception as e:
+        return chat_history, state, gr.update(visible=True), f"API error: {str(e)}"
+
+def image_analysis_fn(image, state):
+    """Analyze uploaded image using OpenAI Responses API."""
+    api_key = state.get("api_key")
+    if not api_key:
+        return "", gr.update(visible=True), "Please enter your OpenAI API key."
+    if image is None:
+        return "Please upload an image.", gr.update(visible=False), ""
+    base64_img = encode_image_to_base64(image)
+    messages = [
+        {
+            "role": "user",
+            "content": [
+                {"type": "input_text", "text": "Please describe this image."},
+                {"type": "input_image", "image_url": f"data:image/jpeg;base64,{base64_img}"}
+            ]
+        }
+    ]
+    try:
+        response = openai_responses_api(api_key, messages)
+        return response.get("output_text", ""), gr.update(visible=False), ""
+    except Exception as e:
+        return f"API error: {str(e)}", gr.update(visible=True), f"API error: {str(e)}"
+
+def export_chat(chat_history):
+    """Export chat history as JSON."""
+    import json
+    chat_json = []
+    for user, bot in chat_history:
+        chat_json.append({"user": user, "bot": bot})
+    return gr.File.update(value=("chat_history.json", json.dumps(chat_json, indent=2)), visible=True)
+
+def show_iframe(space_url):
+    """Show iframe code for embedding."""
+    return get_iframe_code(space_url)
+
+# --- Gradio UI ---
+
+with gr.Blocks(theme=gr.themes.Soft(), css=".footer {text-align: center; font-size: 14px; color: gray;}") as demo:
+    # --- Header ---
+    gr.Markdown(f"<h1 style='text-align:center'>{TITLE}</h1>")
+    with gr.Row():
+        for md, url in LINKS:
+            gr.Markdown(f"<a href='{url}' target='_blank'>{md}</a>")
+    gr.Markdown("---")
+
+    # --- API Key Modal ---
+    state = gr.State({"api_key": None, "previous_response_id": None})
+    with gr.Row(visible=True) as api_key_row:
+        api_key_box = gr.Textbox(label="Enter your OpenAI API Key", type="password", placeholder="sk-...", show_label=True)
+        api_key_btn = gr.Button("Submit")
+        api_key_error = gr.Markdown("", visible=False)
+    def _hide_api_key_row():
+        return gr.update(visible=False)
+    api_key_box.submit(set_api_key, [api_key_box, state], [api_key_row, api_key_error, state])
+    api_key_btn.click(set_api_key, [api_key_box, state], [api_key_row, api_key_error, state])
+
+    # --- Tabs ---
+    with gr.Tabs():
+        # --- Chat Tab ---
+        with gr.Tab("💬 Chatbot"):
+            chatbot = gr.Chatbot(label="Chatbot", height=400)
+            with gr.Row():
+                user_input = gr.Textbox(label="Your message", placeholder="Type your message here...", scale=4)
+                image_input = gr.Image(type="filepath", label="Attach image (optional)", scale=1)
+                send_btn = gr.Button("Send", scale=1)
+            chat_error = gr.Markdown("", visible=False)
+            with gr.Row():
+                export_btn = gr.Button("Export chat as JSON")
+                export_file = gr.File(label="Download chat JSON", visible=False)
+                iframe_btn = gr.Button("Get iframe code")
+                iframe_code = gr.Textbox(label="Embed code", interactive=False, visible=False)
+            # Chat logic
+            send_btn.click(chat_fn, [user_input, chatbot, image_input, state], [chatbot, state, chat_error, api_key_error])
+            user_input.submit(chat_fn, [user_input, chatbot, image_input, state], [chatbot, state, chat_error, api_key_error])
+            export_btn.click(export_chat, chatbot, export_file)
+            iframe_btn.click(show_iframe, gr.Textbox(value="https://huggingface.co/spaces/your-space-name", visible=False), iframe_code)
+        # --- Image Analysis Tab ---
+        with gr.Tab("🖼️ Image Analysis"):
+            image_upload = gr.Image(type="filepath", label="Upload image")
+            analyze_btn = gr.Button("Analyze")
+            image_result = gr.Textbox(label="Analysis result", lines=4)
+            image_error = gr.Markdown("", visible=False)
+            analyze_btn.click(image_analysis_fn, [image_upload, state], [image_result, api_key_error, image_error])
+
+    # --- Theme Toggle ---
+    gr.ThemeSwitcher()
+
+    # --- Footer ---
+    gr.Markdown(f"<div class='footer'>{FOOTER}</div>")
+
+# --- Launch ---
+if __name__ == "__main__":
+    demo.launch()