Update app.py

app.py

@@ -15,8 +15,24 @@ def extract_code(llm_output):
             return code_match.group(1)
         return None
 
-# funcion para prompts
+# funcion para prompts y ejemplo basico
 
+#prompt = "Generate a graph idea based on European fertility data."
+#output = generator(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
+
+# Aqui vamos a añadir los prompts, comparativa entre paises, lo ideal es mas de uno, pero podriamos iniciar en un 1v1 con la metrica 
+# Vamos a generar un codigo para luego ejecutarlo con un exec() y poder imprimir en Streamlit st.pyplot()
+
+def generate_graph_prompt(country1, country2, metric, start_year, end_year):
+    
+    prompt = f"""You have access to a database of European countries with data on {metric}, labor force participation, population, and their predictions for future years.
+        Generate Python code using matplotlib to create a line graph showing the trend of {metric} for {country1} and {country2} from {start_year} to {end_year}.
+        Also, provide a concise explanation of what this graph represents for an end user who might not be familiar with the data.
+        """
+    return prompt
+
+# Ejemplo de como quedaria el prompt que recibiria el modelo
+#prompt = generate_graph_prompt("Germany", "France", "fertility rate", 2020, 2030)
 
 
 #Aqui van las credenciales, conectar las credenciales de Supabase en "Secrets"