Update app.py

app.py

@@ -1,20 +1,52 @@
 import gradio as gr
-from transformers import pipeline
+import requests
+import json
 
-model_name = "meta-llama/Llama-3.2-1B-Instruct"
-pipe = pipeline("text-generation", model=model_name)
+# Replace with your actual API key
+api_key = "YOUR_GEMINI_API_KEY"
 
-def generate_text(prompt):
-    response = pipe(prompt, max_length=200)
-    return response["generated_text"]
+# Define the API endpoint and headers
+url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={api_key}"
+headers = {
+    'Content-Type': 'application/json'
+}
 
+def generate_prompt(prompt):
+    # Prepare the request body
+    data = {
+        "contents": [
+            {
+                "parts": [
+                    {
+                        "text": prompt
+                    }
+                ]
+            }
+        ]
+    }
+
+    # Send the POST request
+    response = requests.post(url, headers=headers, data=json.dumps(data))
+
+    # Check if the request was successful
+    if response.status_code == 200:
+        response_data = response.json()
+        generated_text = response_data['candidates']['content']
+        return generated_text
+    else:
+        print(f"Error: {response.status_code}")
+        print(response.text)
+        return None
+
+# Create the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Meta LLaMA Text Generation")
-    with gr.Tab("Text Generation"):
+    gr.Markdown("# Gemini Prompt Generator")
+    with gr.Tab("Generate Prompt"):
         prompt_input = gr.Textbox(label="Your Prompt:", placeholder="Enter your prompt here", lines=5)
-        generate_button = gr.Button("Generate Text")
-        output_text = gr.Textbox(label="Generated Text:", lines=10, interactive=False)
-        generate_button.click(generate_text, inputs=prompt_input, outputs=output_text)
+        generate_button = gr.Button("Generate Prompt")
+        output_text = gr.Textbox(label="Generated Prompt:", lines=10, interactive=False)
+        generate_button.click(generate_prompt, inputs=prompt_input, outputs=output_text)
 
+# Launch the Gradio application
 if __name__ == "__main__":
-    demo.launch() 
+    demo.launch()