Update app.py

app.py

@@ -1,39 +1,29 @@
 import gradio as gr
 import openai
-from plantuml import PlantUML
-import tempfile
-import os
 
-def generate_plantuml_code(api_key, input_text):
+def generate_plantuml(api_key, text):
     openai.api_key = api_key
     response = openai.ChatCompletion.create(
-        model='gpt-3.5-turbo',
-        messages=[{"role": "user", "content": f"Generate PlantUML code for the following use case: {input_text}"}],
-        max_tokens=150,
-        temperature=0,
+        model="gpt-3.5-turbo",
+        messages=[
+            {
+                "role": "system",
+                "content": "You are ChatGPT, a large language model trained by OpenAI. Generate PlantUML code for the following use case or code in natural language.",
+            },
+            {"role": "user", "content": text},
+        ],
     )
-    plantuml_code = response["choices"][0]["message"]["content"]
-    return plantuml_code
+    return response["messages"][-1]["content"]
 
-def render_plantuml_diagram(plantuml_code):
-    puml = PlantUML(url='http://www.plantuml.com/plantuml/img/')
-    svg_diagram = puml.processes(plantuml_code)
-    return svg_diagram
+iface = gr.Interface(
+    fn=generate_plantuml,
+    inputs=[
+        gr.inputs.Textbox(label="OpenAI API Key"),
+        gr.inputs.Textbox(label="Enter use case or code in natural language")
+    ],
+    outputs=gr.outputs.Code(label="Generated PlantUML Code"),
+    title="PlantUML Code Generator",
+    description="Generate PlantUML code using OpenAI's GPT-3.5-Turbo",
+)
 
-def app(api_key, input_text="", plantuml_code=""):
-    if input_text:
-        plantuml_code = generate_plantuml_code(api_key, input_text)
-    svg_diagram = render_plantuml_diagram(plantuml_code)
-    return plantuml_code, svg_diagram
-
-inputs = [
-    gr.inputs.Textbox(label="Enter OpenAI API Key"),
-    gr.inputs.Textbox(lines=5, label="Enter use case in natural language (optional)"),
-    gr.inputs.Textbox(lines=5, label="Paste PlantUML code (optional)")
-]
-outputs = [
-    gr.outputs.Textbox(label="Generated PlantUML Code"),
-    gr.outputs.HTML(label="Rendered Diagram")
-]
-
-gr.Interface(fn=app, inputs=inputs, outputs=outputs).launch()
+iface.launch()