Update app.py

app.py

@@ -1,46 +1,39 @@
-import os
 import gradio as gr
 import requests
+import os
+from dotenv import load_dotenv
 
-# Set your Black Box API key
-API_KEY = os.getenv("sk-Ro0qnkz8DKLBGsuvsDRPaQ")
+# Load environment variables
+load_dotenv()
+BLACKBOX_API_KEY = os.getenv("sk-Ro0qnkz8DKLBGsuvsDRPaQ")
 
 def generate_text(prompt):
+    # Call the Blackbox AI API
     headers = {
-        "Authorization": f"Bearer {API_KEY}",
+        "Authorization": f"Bearer {BLACKBOX_API_KEY}",
         "Content-Type": "application/json"
     }
     data = {
         "prompt": prompt,
-        "max_tokens": 150
+        "max_tokens": 150,  # Adjust as needed
+        "temperature": 0.7   # Adjust as needed
     }
-    response = requests.post("https://api.blackbox.com/generate", headers=headers, json=data)
+    
+    response = requests.post("https://api.blackbox.ai/generate", headers=headers, json=data)
     
     if response.status_code == 200:
-        return response.json().get("text", "No text generated.")
+        return response.json().get("text", "No response text found.")
     else:
-        return "Error: " + response.text
-
-def suggest_improvement(text):
-    # Placeholder for suggestion logic
-    return f"Suggestion for improvement: {text} could be enhanced by adding more details."
-
-with gr.Blocks() as demo:
-    gr.Markdown("## Text Generation and Improvement Suggestions")
-    
-    with gr.Row():
-        with gr.Column():
-            prompt_input = gr.Textbox(label="Enter your prompt", placeholder="Type your text here...")
-            generate_button = gr.Button("Generate Text")
-            output_text = gr.Textbox(label="Generated Text", interactive=False)
-        
-        with gr.Column():
-            improvement_input = gr.Textbox(label="Enter text for suggestions", placeholder="Type your text here...")
-            suggest_button = gr.Button("Get Suggestions")
-            suggestion_output = gr.Textbox(label="Improvement Suggestions", interactive=False)
+        return f"Error: {response.status_code} - {response.text}"
 
-    generate_button.click(generate_text, inputs=prompt_input, outputs=output_text)
-    suggest_button.click(suggest_improvement, inputs=improvement_input, outputs=suggestion_output)
+# Create Gradio interface
+iface = gr.Interface(
+    fn=generate_text,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
+    outputs="text",
+    title="Text-to-Text Copilot",
+    description="A copilot for text generation that learns from coding patterns and provides suggestions."
+)
 
 if __name__ == "__main__":
-    demo.launch()
+    iface.launch()