Convert to Gradiog

Procfile.txt

@@ -0,0 +1 @@
+web: uvicorn app:app --host 0.0.0.0 --port $PORT

app.py

@@ -1,55 +1,25 @@
 import os
 import logging
-from fastapi import FastAPI, Request
-from pydantic import BaseModel
-from fastapi.responses import JSONResponse
 import httpx
 from dotenv import load_dotenv
+import gradio as gr
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Load environment variables from .env file
+# Load environment variables
 load_dotenv()
 logger.info("Environment variables loaded from .env file")
 logger.info(f"OPENAI_API_KEY present: {'OPENAI_API_KEY' in os.environ}")
 logger.info(f"ANTHROPIC_API_KEY present: {'ANTHROPIC_API_KEY' in os.environ}")
 logger.info(f"GEMINI_API_KEY present: {'GEMINI_API_KEY' in os.environ}")
 
-app = FastAPI()
-
-class QueryRequest(BaseModel):
-    query: str
-    provider: str  # e.g., "openai", "anthropic", "gemini"
-
-@app.get("/")
-def read_root():
-    return {"message": "Multi-Model Selector is running"}
-
-@app.post("/ask")
-async def ask_question(request: QueryRequest):
-    query = request.query
-    provider = request.provider.lower()
-
-    try:
-        if provider == "openai":
-            return await ask_openai(query)
-        elif provider == "anthropic":
-            return await ask_anthropic(query)
-        elif provider == "gemini":
-            return await ask_gemini(query)
-        else:
-            return JSONResponse(content={"error": f"Unknown provider: {provider}"}, status_code=400)
-    except Exception as e:
-        logger.error(f"Error processing request: {str(e)}")
-        return JSONResponse(content={"error": str(e)}, status_code=500)
-
 async def ask_openai(query: str):
     openai_api_key = os.getenv("OPENAI_API_KEY")
     if not openai_api_key:
         logger.error("OpenAI API key not provided")
-        return {"error": "OpenAI API key not provided."}
+        return "Error: OpenAI API key not provided."
 
     headers = {
         "Authorization": f"Bearer {openai_api_key}",
@@ -67,20 +37,20 @@ async def ask_openai(query: str):
         
         response.raise_for_status()
         answer = response.json()['choices'][0]['message']['content']
-        return {"response": answer}
+        return answer
 
     except httpx.HTTPStatusError as e:
         logger.error(f"OpenAI HTTP Status Error: {e.response.status_code}, {e.response.text}")
-        return {"error": f"OpenAI HTTP Status Error: {e.response.status_code}, {e.response.text}"}
+        return f"Error: OpenAI HTTP Status Error: {e.response.status_code}, {e.response.text}"
     except Exception as e:
         logger.error(f"OpenAI Error: {str(e)}")
-        return {"error": f"OpenAI Error: {str(e)}"}
+        return f"Error: OpenAI Error: {str(e)}"
 
 async def ask_anthropic(query: str):
     anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
     if not anthropic_api_key:
         logger.error("Anthropic API key not provided")
-        return {"error": "Anthropic API key not provided."}
+        return "Error: Anthropic API key not provided."
 
     headers = {
         "x-api-key": anthropic_api_key,
@@ -102,20 +72,20 @@ async def ask_anthropic(query: str):
         response.raise_for_status()
         logger.info(f"Anthropic response: {response.json()}")
         answer = response.json()['content'][0]['text']
-        return {"response": answer}
+        return answer
 
     except httpx.HTTPStatusError as e:
         logger.error(f"Anthropic HTTP Status Error: {e.response.status_code}, {e.response.text}")
-        return {"error": f"Anthropic HTTP Status Error: {e.response.status_code}, {e.response.text}"}
+        return f"Error: Anthropic HTTP Status Error: {e.response.status_code}, {e.response.text}"
     except Exception as e:
         logger.error(f"Anthropic Error: {str(e)}")
-        return {"error": f"Anthropic Error: {str(e)}"}
+        return f"Error: Anthropic Error: {str(e)}"
 
 async def ask_gemini(query: str):
     gemini_api_key = os.getenv("GEMINI_API_KEY")
     if not gemini_api_key:
         logger.error("Gemini API key not provided")
-        return {"error": "Gemini API key not provided."}
+        return "Error: Gemini API key not provided."
 
     headers = {
         "Content-Type": "application/json"
@@ -135,11 +105,41 @@ async def ask_gemini(query: str):
         
         response.raise_for_status()
         answer = response.json()['candidates'][0]['content']['parts'][0]['text']
-        return {"response": answer}
+        return answer
 
     except httpx.HTTPStatusError as e:
         logger.error(f"Gemini HTTP Status Error: {e.response.status_code}, {e.response.text}")
-        return {"error": f"Gemini HTTP Status Error: {e.response.status_code}, {e.response.text}"}
+        return f"Error: Gemini HTTP Status Error: {e.response.status_code}, {e.response.text}"
     except Exception as e:
         logger.error(f"Gemini Error: {str(e)}")
-        return {"error": f"Gemini Error: {str(e)}"}
+        return f"Error: Gemini Error: {str(e)}"
+
+async def query_model(query: str, provider: str):
+    provider = provider.lower()
+    if provider == "openai":
+        return await ask_openai(query)
+    elif provider == "anthropic":
+        return await ask_anthropic(query)
+    elif provider == "gemini":
+        return await ask_gemini(query)
+    else:
+        return f"Error: Unknown provider: {provider}"
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Multi-Model Selector")
+    gr.Markdown("Select a provider and enter a query to get a response from the chosen AI model.")
+    
+    provider = gr.Dropdown(choices=["OpenAI", "Anthropic", "Gemini"], label="Select Provider")
+    query = gr.Textbox(label="Enter your query", placeholder="e.g., What is the capital of the United States?")
+    submit_button = gr.Button("Submit")
+    output = gr.Textbox(label="Response", interactive=False)
+    
+    submit_button.click(
+        fn=query_model,
+        inputs=[query, provider],
+        outputs=output
+    )
+
+# Launch the Gradio app
+demo.launch()

requirements.txt

@@ -5,3 +5,7 @@ google-generativeai
 anthropic
 python-dotenv
 requests
+gradio
+httpx
+
+