Update app.py

app.py

@@ -47,24 +47,8 @@ def gpt_inference(base_url, model, openai_key, prompt):
     }
 
     response = requests.post(f"{base_url}/v1/chat/completions", headers=headers, data=json.dumps(data))
-
-    def extract_code(text):
-        # Match triple backtick blocks first
-        triple_match = re.search(r'```(?:\w+\n)?(.+?)```', text, re.DOTALL)
-        if triple_match:
-            return triple_match.group(1).strip()
-        else:
-            # If no triple backtick blocks, match single backtick blocks
-            single_match = re.search(r'`(.+?)`', text, re.DOTALL)
-            if single_match:
-                return single_match.group(1).strip()
-        # If no code blocks found, return original text
-        return text
-
-    if response.status_code != 200:
-        return "Error: " + response.text, 500
     
-    code = extract_code(response.json()["choices"][0]["message"]["content"])
+    code = response.json()["choices"][0]["message"]["content"]
     print(f"code:{code}")
     img = execute_code(code)
     return img