Add prompt and llm code

app.py

@@ -0,0 +1,67 @@
+import gradio as gr
+import spaces
+import os
+from mistralai import Mistral, UserMessage, SystemMessage
+
+# Get the API key and endpoint
+token = os.environ.get("GITHUB_TOKEN")
+endpoint = "https://models.github.ai/inference"
+model_name = "mistral-ai/Codestral-2501"
+
+# Initialize the Mistral client
+client = Mistral(api_key=token, server_url=endpoint)
+
+def generate_response(query, diagram_type):
+    if diagram_type == "ERD":
+        system_prompt = (
+            "You are a code generation expert. Convert the following PlantUML code into "
+            "equivalent SQL code. Give only code as output."
+        )
+    elif diagram_type == "Use Case Diagram":
+        system_prompt = (
+            "You are a code generation expert. Convert the following PlantUML code into "
+            "equivalent Java REST API controller methods."
+        )
+    elif diagram_type in ["Class Diagram", "Sequence Diagram"]:
+        system_prompt = (
+            "You are a code generation expert. Convert the following PlantUML code into "
+            "equivalent Java code. Give only code as output."
+        )
+    else:
+        system_prompt = "You are a helpful coding assistant."
+
+    user_prompt = query.strip()
+
+    response = client.chat.complete(
+        model=model_name,
+        messages=[
+            SystemMessage(system_prompt),
+            UserMessage(user_prompt),
+        ],
+        max_tokens=1000,
+        model_name=model_name,
+    )
+
+    return response.choices[0].message.content
+
+# ---------------- Gradio Interface ----------------
+
+@spaces.GPU
+def predict(query, diagram_type):
+    return generate_response(query, diagram_type)
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=[
+        gr.Textbox(lines=15, label="PlantUML Code"),
+        gr.Dropdown(
+            choices=["ERD", "Use Case Diagram", "Class Diagram", "Sequence Diagram"],
+            label="Diagram Type"
+        )
+    ],
+    outputs="text",
+    title="PlantUML-To-Code Converter",
+    description="Enter PlantUML code to generate equivalent SQL or Java code based on the diagram type."
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio>=4.0.0
+spaces>=0.16.0
+mistralai>=1.0.0