Update app.py

app.py

@@ -150,8 +150,8 @@ with st.sidebar:
         "Comuni",
         elenco
     )
-    prezzo_minimo = st.sidebar.slider("Prezzo Minimo", min_value=0, max_value=1000, value=200)
-    prezzo_massimo = st.sidebar.slider("Prezzo Massimo", min_value=0, max_value=1000, value=230)
+    prezzo_minimo = st.sidebar.slider("Prezzo Minimo (k)", min_value=0, max_value=1000, value=200)
+    prezzo_massimo = st.sidebar.slider("Prezzo Massimo (k)", min_value=0, max_value=1000, value=230)
     locali = list(range(1, 21))  # Intervallo da 1 a 10
     locali_range = st.sidebar.select_slider(
         "Locali",
@@ -159,6 +159,8 @@ with st.sidebar:
         value=(locali[2], locali[4])  # Valore iniziale, da 1 a 5 locali
     )
     mostra_grafici = st.toggle("Mostra grafici", value = True)
+    
+    modello_ai = st.selectbox("Modello A.I.", ("Llama 3.1", "Mixtral 7x8b"))
     analisi_ai = st.toggle("Analizza i dati tramite l'A.I.", value = True)
 
     locali_minimo, locali_massimo = locali_range
@@ -235,36 +237,71 @@ def scrivi_dataframe(output, riepilogo):
         st.write(f"Riepilogando nel comune {comune_provincia['comune']} sono presenti **{vantaggioso_count} Immobili vantaggiosi** rispetto ai {total_rows} totali")
         st.divider()
 
-
-def  analizza_dati_ai(output):
-    Ak = "XXXXgsk_COQPhsuaXqvuCkKPryiQWGdyb3FYCHBSIcn30wk3ZZay8WYC6tpIXXXX"
+def  analizza_dati_ai(output, tipologia):
+    Ak = "XXgsk_COQPhsuaXqvuCkKPryiQWGdyb3FYCHBSIcn30wk3ZZay8WYC6tpIXX"
     client = Groq(api_key=Ak)
-    prompt_messages = [
-        {
-            "role": "system",
-            "content": "Sei un'assistente virtuale specializzata nel supporto agli investitori immobiliari. Il tuo compito è analizzare i dati di un file CSV contenente un elenco di immobili e fornire consigli, idee e analisi dettagliate per massimizzare il profitto attraverso la riqualificazione e rivendita degli stessi. Devi considerare vari fattori come la posizione degli immobili, il prezzo di acquisto, il potenziale valore di mercato post-riqualificazione, i costi stimati per la ristrutturazione e altri dati pertinenti. Il tuo obiettivo è offrire suggerimenti strategici basati sui dati, evidenziare le migliori opportunità di investimento, identificare eventuali rischi, e fornire consigli pratici per ottimizzare i profitti. Rispondi in modo chiaro, conciso e professionale."
-        },
-        {
-            "role": "user",
-            "content": f"Ho un file CSV con un elenco di immobili che sto considerando per un investimento. Vorrei che tu analizzassi i dati e mi fornissi consigli dettagliati su come posso massimizzare il mio profitto attraverso la riqualificazione e la rivendita di questi immobili. Per ogni immobile, per favore prendi in considerazione i seguenti aspetti:\n\n1. Valore attuale: Quanto vale l'immobile al momento dell'acquisto?\n2. Costi di riqualificazione stimati: Quanto potrebbe costare la ristrutturazione o la riqualificazione?\n3. Valore di mercato potenziale: Quanto potrebbe valere l'immobile una volta riqualificato?\n4. Rendimento potenziale: Qual è il margine di profitto previsto, tenendo conto dei costi totali?\n5. Tempistica di rivendita: In quanto tempo è probabile che l'immobile venga venduto una volta riqualificato?\n6. Analisi del mercato locale: Qual è la situazione del mercato immobiliare nella zona specifica? Ci sono trend emergenti?\n7. Rischi potenziali: Quali sono i possibili rischi o svantaggi legati a ciascun immobile?\nInfine, suggeriscimi le migliori opportunità di investimento tra gli immobili elencati, con una breve spiegazione del motivo per cui questi immobili sono i più promettenti.\n\n{output}"  
-        }
-    ]
+    if tipologia == 1:
+        prompt_messages = [
+            {
+                "role": "system",
+                "content": """Sei un'assistente virtuale specializzata nel supporto agli investitori immobiliari. 
+                            Il tuo compito è analizzare i dati di un file CSV contenente un elenco di immobili e fornire consigli, idee e analisi dettagliate 
+                            per massimizzare il profitto attraverso la riqualificazione e rivendita degli stessi. 
+                            Devi considerare vari fattori come la posizione degli immobili, il prezzo di acquisto, il potenziale valore di mercato post-riqualificazione, 
+                            i costi stimati per la ristrutturazione e altri dati pertinenti. 
+                            Il tuo obiettivo è offrire suggerimenti strategici basati sui dati, evidenziare le migliori opportunità di investimento, 
+                            identificare eventuali rischi, e fornire consigli pratici per ottimizzare i profitti. Rispondi in modo chiaro, conciso e professionale."""
+            },
+            {
+                "role": "user",
+                "content": f"""Ho un file CSV con un elenco di immobili che sto considerando per un investimento. 
+                            Vorrei che tu analizzassi i dati e mi fornissi consigli dettagliati su come posso massimizzare il mio profitto attraverso la riqualificazione e la rivendita di questi immobili. 
+                            Per ogni immobile, per favore prendi in considerazione i seguenti aspetti:\n\n
+                            1. Valore attuale: Quanto vale l'immobile al momento dell'acquisto?\n
+                            2. Costi di riqualificazione stimati: Quanto potrebbe costare la ristrutturazione o la riqualificazione?\n
+                            3. Valore di mercato potenziale: Quanto potrebbe valere l'immobile una volta riqualificato?\n
+                            4. Rendimento potenziale: Qual è il margine di profitto previsto, tenendo conto dei costi totali?\n
+                            5. Tempistica di rivendita: In quanto tempo è probabile che l'immobile venga venduto una volta riqualificato?\n
+                            6. Analisi del mercato locale: Qual è la situazione del mercato immobiliare nella zona specifica? Ci sono trend emergenti?\n
+                            7. Rischi potenziali: Quali sono i possibili rischi o svantaggi legati a ciascun immobile?\n
+                            Infine, suggeriscimi le migliori opportunità di investimento tra gli immobili elencati, 
+                            con una breve spiegazione del motivo per cui questi immobili sono i più promettenti.
+                            DATI UN PUNTEGGIO DA 0 A 100 e mostra i primi 5 in ordine di punteggio DECRESCENTE!
+                            \n\n{output}"""
+            }
+        ]
+    elif tipologia==2:
+        prompt_messages = [
+            {
+                "role": "system",
+                "content": """Sei un'assistente virtuale specializzata nel supporto agli investitori immobiliari"""
+            },
+            {
+                "role": "user",
+                "content": f"""Ho un file CSV con un elenco di immobili che sto considerando per un investimento. 
+                               Analizza i dati fornendo un punteggio da 1 a 100 degli immobili specificando PRO e CONTRO per ognuno. Mettili in ORDINE di punteggio DECRESCENTE. Restituisci un CSV formattato\n\n{output}"""
+            }
+        ]
+    print(prompt_messages)
+    
     response_area = st.empty()
-    completion = client.chat.completions.create(
-        model="llama-3.1-70b-versatile",
-        messages=prompt_messages,
-        temperature=1,
-        max_tokens=1024,
-        top_p=1,
-        stream=True,
-        stop=None,
-    )
-    response_text = ""
-    for chunk in completion:
-        response_text += chunk.choices[0].delta.content or ""
-        response_area.text(response_text)    
+    with st.spinner("Generazione in corso..."):
+        completion = client.chat.completions.create(
+            model="llama-3.1-70b-versatile",
+            messages=prompt_messages,
+            temperature=1,
+            max_tokens=2048,
+            top_p=1,
+            stream=True,
+            stop=None,
+        )
+        response_text = ""
+        for chunk in completion:
+            response_text += chunk.choices[0].delta.content or ""
+            response_area.markdown(response_text + "▌")
+        response_area.markdown(response_text)
 
-if cerca_premuto: 
+if cerca_premuto:
     if len(comune_input)>0:
         comuni_selezionati = comune_input
         comuni_selezionati = [comune.upper() for comune in comuni_selezionati]
@@ -286,7 +323,12 @@ if cerca_premuto:
             st.write(f"### Comuni Selezionati")
             scrivi_dataframe(output, True)
         if st.numero_immobili_validi > 0 and analisi_ai:  
-            analizza_dati_ai(output)        
+            st.title("✨ Analisi Intelligenza Artificiale")
+            st.write("### Considerazioni")
+            analizza_dati_ai(output, 1)
+            #st.divider()
+            #st.write("### Dettagliata")
+            #analizza_dati_ai(output, 2)
         st.success("Elaborazione Completata")
     else:
-        st.error("Per favore, inserisci il nome di un comune.")  
+        st.error("Per favore, inserisci il nome di un comune.")