saicharantej commited on
Commit
d4c7c80
·
1 Parent(s): 4d5da8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -12
app.py CHANGED
@@ -1,6 +1,11 @@
1
  import os
2
  os.environ["OPENAI_API_KEY"] = "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"
3
  import openai
 
 
 
 
 
4
  def get_completion(prompt, model="gpt-3.5-turbo"):
5
  messages = [{"role": "user", "content": prompt}]
6
  response = openai.ChatCompletion.create(
@@ -25,13 +30,6 @@ def extract_text_between_backticks(text):
25
  pattern = r"```(.+?)```"
26
  matches = re.findall(pattern, text, re.DOTALL)
27
  return matches
28
- import gradio as gr
29
- from PIL import Image
30
-
31
- import base64
32
- import requests
33
-
34
-
35
 
36
  def generate_diagram(mermaid_code):
37
  graphbytes = mermaid_code.encode("ascii")
@@ -50,11 +48,6 @@ def visualize_diagram(input_text):
50
 
51
  input_text = gr.inputs.Textbox(label="Enter your question")
52
  output_image = gr.outputs.Image(type="pil", label="Diagram")
53
- examples = [
54
- ["Explain the architecture of Twitter"],
55
- ["How does Oauth work?"],
56
- ["Explain how Global warming affects life on earth"]
57
- ]
58
 
59
  gr.Interface(
60
  fn=visualize_diagram,
 
1
  import os
2
  os.environ["OPENAI_API_KEY"] = "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"
3
  import openai
4
+ import gradio as gr
5
+ from PIL import Image
6
+ import base64
7
+ import requests
8
+
9
  def get_completion(prompt, model="gpt-3.5-turbo"):
10
  messages = [{"role": "user", "content": prompt}]
11
  response = openai.ChatCompletion.create(
 
30
  pattern = r"```(.+?)```"
31
  matches = re.findall(pattern, text, re.DOTALL)
32
  return matches
 
 
 
 
 
 
 
33
 
34
  def generate_diagram(mermaid_code):
35
  graphbytes = mermaid_code.encode("ascii")
 
48
 
49
  input_text = gr.inputs.Textbox(label="Enter your question")
50
  output_image = gr.outputs.Image(type="pil", label="Diagram")
 
 
 
 
 
51
 
52
  gr.Interface(
53
  fn=visualize_diagram,