Update app.py

app.py

@@ -46,10 +46,37 @@ def generate_mermaid_code(prompt):
     if response.status_code == 200:
         result = response.json()
         # Extract the generated Mermaid.js code
-        return result[0]['generated_text'].strip()
+        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):
+
+    mermaid_prompt = f"{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
+        return result[0]['generated_text'].strip()
+    else:
+        return f"Error: {response.status_code}, {response.text}"    
+
 # Function to process text, audio, or both inputs
 def process_input(input_type, text_input, audio_input):
     if input_type == "Audio" and audio_input is not None: