Spaces:

harris1
/

Codestralhack

Building

App Files Files Community

harris1 commited on Jul 17

Commit

e4daf67

•

1 Parent(s): ae0d257

Update app.py

Browse files

Files changed (1) hide show

app.py +15 -18

app.py CHANGED Viewed

@@ -285,12 +285,12 @@
 # if __name__ == "__main__":
 #     iface.launch()
-from flask import Flask, request, jsonify, render_template_string
 import os
 from mistralai.client import MistralClient
 from mistralai.models.chat_completion import ChatMessage
-# Ensure the environment variable for the API key is set
 api_key = os.getenv("MISTRAL_API_KEY")
 if not api_key:
     raise ValueError("MISTRAL_API_KEY environment variable not set")
@@ -298,20 +298,17 @@ if not api_key:
 model = "mistral-tiny"
 client = MistralClient(api_key=api_key)
-app = Flask(__name__)
 def generate_goals(input_var):
     messages = [
         ChatMessage(role="user", content=f"Generate 5 specific, industry relevant goals for {input_var} using Python and Pandas in exam data analysis. Each goal should include a brief name and a one-sentence description of the task or skill.")
     ]
     try:
         response = client.chat(model=model, messages=messages)
-        content = response.choices[0].message.content
-        return content
     except Exception as e:
         return f"An error occurred: {str(e)}"
 # HTML content with interactive visualization
 html_content = """
 <!DOCTYPE html>
@@ -511,15 +508,15 @@ html_content = """
 </html>
 """
-@app.route('/')
-def index():
-    return render_template_string(html_content)
-@app.route('/generate_goals', methods=['POST'])
-def generate_goals_api():
-    input_var = request.json['input_var']
-    goals = generate_goals(input_var)
-    return jsonify({'goals': goals})
-if __name__ == "__main__":
-    app.run(host='0.0.0.0', port=5000, debug=True)

 # if __name__ == "__main__":
 #     iface.launch()
+import gradio as gr
 import os
 from mistralai.client import MistralClient
 from mistralai.models.chat_completion import ChatMessage
+# Mistral AI setup
 api_key = os.getenv("MISTRAL_API_KEY")
 if not api_key:
     raise ValueError("MISTRAL_API_KEY environment variable not set")
 model = "mistral-tiny"
 client = MistralClient(api_key=api_key)
 def generate_goals(input_var):
     messages = [
         ChatMessage(role="user", content=f"Generate 5 specific, industry relevant goals for {input_var} using Python and Pandas in exam data analysis. Each goal should include a brief name and a one-sentence description of the task or skill.")
     ]
     try:
         response = client.chat(model=model, messages=messages)
+        return response.choices[0].message.content
     except Exception as e:
         return f"An error occurred: {str(e)}"
 # HTML content with interactive visualization
 html_content = """
 <!DOCTYPE html>
 </html>
 """
+# Gradio interface
+iface = gr.Interface(
+    fn=generate_goals,
+    inputs="text",
+    outputs="text",
+    title="Exam Data Analysis Goals Generator",
+    description="Click on a goal in the visualization to generate related goals.",
+    article=html_content
+)
+# Launch the Gradio app
+iface.launch()