import google.generativeai as genai
import random
import time
import gradio as gr

API_KEY = 'AIzaSyAd4Nl2ft8txIZnK382b1gHv6dsPTtdrwI'
genai.configure(api_key=API_KEY)
model = genai.GenerativeModel('gemini-pro')

def generate_idea():
    response = model.generate_content("Write any idea for python code. Give response as only this idea")
    response.resolve()
    idea = response.text
    return idea

def generate_code(idea):
    response = model.generate_content(f"Write code for {idea}. As response give only code")
    response.resolve()
    code = response.text
    return code

def generate_code_with_idea(num_ideas=1):
    output = ""
    for i in range(num_ideas):
        idea = generate_idea()
        code = generate_code(idea)
        output += f"Idea {i+1}: {idea}\n\nCode:\n{code}\n\n---\n\n"
    return output

iface = gr.Interface(
    fn=generate_code_with_idea,
    inputs=gr.Slider(1, 10, 1, label="Number of Ideas"),
    outputs="text",
    title="Python Code Idea Generator",
    description="Generate Python code ideas and code snippets using Gemini Pro."
)

iface.launch()