File size: 1,585 Bytes
ea3ab8f 3e310fc 63d9109 ea3ab8f 9ed3b28 665d0ad 3e310fc 665d0ad 3e310fc 665d0ad 3e310fc 665d0ad 3e310fc 665d0ad 3e310fc 665d0ad 4884c35 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import gradio as gr
import requests
import json
import os
api_key = os.environ.get("AIzaSyBC0lwB4yx-c0C94y7ENnQEzgMHSyqM3l0")
# 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("# 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 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(share = True) |