Update app.py

app.py

@@ -24,35 +24,35 @@ def transcribe_audio(audio_file):
         return f"Error: {response.status_code}, {response.text}"
 
 # Function to query the Mistral model to generate Mermaid.js code
-def generate_mermaid_code(prompt):
-    # mermaid_prompt = f"Use the appropriate diagram type (Use Case Diagram, Flowchart, Sequence Diagram, Entity-Relationship (ER) Diagram,State Diagram, Pie Chart etc.) based on the context.\n" \
-    #                  "Generate a valid, syntactically correct MermaidJS diagram code for the following: {prompt}"
-    mermaid_prompt = f"Generate a valid MermaidJS diagram code for the following: {prompt}"
-
-
-    # Prepare the payload (input for the model)
-    payload = {
-        "inputs": mermaid_prompt,
-        "parameters": {
-            "max_length": 256,
-            "temperature": 0.7
-        }
-    }
-
-    # Send the request to the Mistral API
-    response = requests.post(MISTRAL_API_URL, headers=headers, json=payload)
+# def generate_mermaid_code(prompt):
+#     # mermaid_prompt = f"Use the appropriate diagram type (Use Case Diagram, Flowchart, Sequence Diagram, Entity-Relationship (ER) Diagram,State Diagram, Pie Chart etc.) based on the context.\n" \
+#     #                  "Generate a valid, syntactically correct MermaidJS diagram code for the following: {prompt}"
+#     mermaid_prompt = f"Generate a valid MermaidJS diagram code for the following: {prompt}"
+
+
+#     # Prepare the payload (input for the model)
+#     payload = {
+#         "inputs": mermaid_prompt,
+#         "parameters": {
+#             "max_length": 256,
+#             "temperature": 0.7
+#         }
+#     }
+
+#     # Send the request to the Mistral API
+#     response = requests.post(MISTRAL_API_URL, headers=headers, json=payload)
+
+#     # Check if the request was successful
+#     if response.status_code == 200:
+#         result = response.json()
+#         # Extract the generated Mermaid.js code
+#         PROMPT1 = f"ORGANISE, OPTIME AND FORMATE MERMAID.JS CODE AND GIVE ME only MERMAID.JS CODE : {result[0]['generated_text'].strip()}" 
+#         return generate_mermaid_code1(PROMPT1)
+#         # return result[0]['generated_text'].strip()
+#     else:
+#         return f"Error: {response.status_code}, {response.text}"
 
-    # Check if the request was successful
-    if response.status_code == 200:
-        result = response.json()
-        # Extract the generated Mermaid.js code
-        PROMPT1 = f"ORGANISE, OPTIME AND FORMATE MERMAID.JS CODE AND GIVE ME only MERMAID.JS CODE : {result[0]['generated_text'].strip()}" 
-        return generate_mermaid_code1(PROMPT1)
-        # return result[0]['generated_text'].strip()
-    else:
-        return f"Error: {response.status_code}, {response.text}"
-
-def generate_mermaid_code1(prompt):
+def generate_mermaid_code(prompt):
 
     # Model endpoint
     MODEL = "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
@@ -65,10 +65,10 @@ def generate_mermaid_code1(prompt):
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
     }
-
+    mermaid_prompt = f"Generate all valid MermaidJS code for the following: {prompt}"
     # Define input messages
     messages = [
-        {"role": "user", "content": prompt}
+        {"role": "user", "content": mermaid_prompt}
     ]
     
     # Extract user content as input text