Kvikontent commited on
Commit
a4a5bfc
·
verified ·
1 Parent(s): 3f6b6ce

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import google.generativeai as genai
2
+ import random
3
+ import time
4
+ import gradio as gr
5
+
6
+ API_KEY = 'AIzaSyAd4Nl2ft8txIZnK382b1gHv6dsPTtdrwI'
7
+ genai.configure(api_key=API_KEY)
8
+ model = genai.GenerativeModel('gemini-pro')
9
+
10
+ def generate_idea():
11
+ response = model.generate_content("Write any idea for python code. Give response as only this idea")
12
+ response.resolve()
13
+ idea = response.text
14
+ return idea
15
+
16
+ def generate_code(idea):
17
+ response = model.generate_content(f"Write code for {idea}. As response give only code")
18
+ response.resolve()
19
+ code = response.text
20
+ return code
21
+
22
+ def generate_code_with_idea(num_ideas=1):
23
+ output = ""
24
+ for i in range(num_ideas):
25
+ idea = generate_idea()
26
+ code = generate_code(idea)
27
+ output += f"Idea {i+1}: {idea}\n\nCode:\n{code}\n\n---\n\n"
28
+ return output
29
+
30
+ iface = gr.Interface(
31
+ fn=generate_code_with_idea,
32
+ inputs=gr.Slider(1, 10, 1, label="Number of Ideas"),
33
+ outputs="text",
34
+ title="Python Code Idea Generator",
35
+ description="Generate Python code ideas and code snippets using Gemini Pro."
36
+ )
37
+
38
+ iface.launch()