Update app.py

app.py

@@ -361,6 +361,10 @@ def create_cluster_dataframes(processed_df):
 from transformers import GPTNeoForCausalLM, GPT2Tokenizer
 def generate_project_proposal(problem_descriptions, location, problem_domain):
     print("Trying to access gpt-neo-1.3B")
+    print("problem_descriptions: ", problem_descriptions)
+    print("location: ", location)
+    print("problem_domain: ", problem_domain)
+    
     model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
     tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
 
@@ -371,17 +375,32 @@ def generate_project_proposal(problem_descriptions, location, problem_domain):
     prompt = f"Generate a solution oriented project proposal for the following:\n\nLocation: {location}\nProblem Domain: {problem_domain}\nProblems: {problems_summary}\n\nProject Proposal:"
 
     # Generate the proposal
-    input_ids = tokenizer.encode(prompt, return_tensors="pt")
-    output = model.generate(
-        input_ids, 
-        max_length=300,
-        num_return_sequences=1, 
-        no_repeat_ngram_size=2,
-        temperature=0.75)
-
-    proposal = tokenizer.decode(output[0], skip_special_tokens=True)
-    print("Successfully accessed gpt-neo-1.3B and returning")
-    return proposal
+    try:
+        input_ids = tokenizer.encode(prompt, return_tensors="pt")
+        print("Input IDs shape:", input_ids.shape)
+        output = model.generate(
+            input_ids, 
+            max_length=300,
+            num_return_sequences=1, 
+            no_repeat_ngram_size=2,
+            temperature=0.75)
+        print("Output shape:", output.shape)
+        proposal = tokenizer.decode(output[0], skip_special_tokens=True)
+        print("Successfully accessed gpt-neo-1.3B and returning")
+        return proposal
+    except Exception as e:
+        print("Error generating proposal:", str(e))
+        return prompt
+
+
+
+
+
+
+
+
+
+    
 
 def create_project_proposals(budget_cluster_df, problem_cluster_df, location_clusters, problem_clusters):
     print("\n Starting function: create_project_proposals")